lyst / lightfm

A Python implementation of LightFM, a hybrid recommendation algorithm.
Apache License 2.0
4.73k stars 691 forks source link

[Feature request] NDCG metric for model evaluation #655

Open jpzhangvincent opened 2 years ago

jpzhangvincent commented 2 years ago

I believe NDCG@k is usually better metric to evaluate a recommendation system. Can we add the NDCG@k metric for model evaluation? Or is there an easy way to calculate the metric based on variables in the model class? Thanks a lot!