twhin: translation embeddings not being added for negative examples

Author: ksanjeevanCreated Jul 16, 2024Updated Jul 16, 2024

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

Source: twitter/the-algorithm-ml