search

hand10ryo / PyTorchCML

PyTorchCML is a library of PyTorch implementations of matrix factorization (MF) and collaborative metric learning (CML), algorithms used in recommendation systems and data mining.
MIT License
20 stars 2 forks source link

User and Item features? #33

Closed shivamtundele closed 2 years ago

shivamtundele commented 2 years ago

Hi! Is there a way to add user and item features to the model just like LightFM?

hand10ryo commented 2 years ago

Thanks for the asking! Now we can build user and item features into the model by domain adoptation. This technique is archived by adding to the loss the distance between the projection from that user/item feature to the embedding space and the embedding of the user/item.

For details, see the README on "adoptors" and section 3.3 of the paper.

shivamtundele commented 2 years ago

Thank you! I would love to contribute on this repo to do that. Are there any contribution guidelines?

hand10ryo commented 2 years ago

Oh! I am so glad to hear that! Here are some quick guidelines for you! Please feel free to comment if you have any questions.