PyTorchCML icon indicating copy to clipboard operation
PyTorchCML copied to clipboard

Any way to get user-user, item-item and user-item embeddings back from trained model?

Open shivamtundele opened this issue 3 years ago • 1 comments

Hello, like LightFM algorithm, is it possible to retrieve user-user, item-item and user-item embeddings back from the trained model?

shivamtundele avatar Sep 08 '22 18:09 shivamtundele

PyTorchCML does not have a mechanism to retrieve. You can retrieve the embedding vector by model.user_embeddings.weight or model.item_embeddings.weight. One of the easiest ways to achieve retrieving would be to enter it into another library such as Annoy.

hand10ryo avatar Sep 26 '22 13:09 hand10ryo