Feature/get topk items - Githubissues

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

Feature/get topk items #37

Closed hand10ryo closed 2 years ago

hand10ryo commented 2 years ago

adding a method to predict top k recommendations for each users https://github.com/hand10ryo/PyTorchCML/issues/36 .

For detail,

add a method to PyTorchCML/models/BaseEmbeddingModel.py
add a test for the method (tests/models/test_CollaborativeMetricLearning.py)
add a example to get topk recommend items (examples/notebooks/movielens_cml.ipynb)

(please ignore other file changes)

shivamtundele commented 2 years ago

I would suggest to add Calendar Versioning for repo releases.