PyTorchCML
PyTorchCML copied to clipboard
Any way to get user-user, item-item and user-item embeddings back from trained model?
Hello, like LightFM algorithm, is it possible to retrieve user-user, item-item and user-item embeddings back from the trained model?
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.