the-algorithm-ml icon indicating copy to clipboard operation
the-algorithm-ml copied to clipboard

twhin: translation embeddings not being added for negative examples

Open ksanjeevan opened this issue 7 months ago • 0 comments

how come only the real edges get the lhs + translation? translated = x[:, 1, :] + trans_embs occurs before the negative examples are generated but translated is not used in the for loop.

https://github.com/twitter/the-algorithm-ml/blob/b85210863f7a94efded0ef5c5ccf4ff42767876c/projects/twhin/models/models.py#L54

ksanjeevan avatar Jul 16 '24 20:07 ksanjeevan