datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
MIT License
400 stars 74 forks source link

(Need Review) Change the calculation of hit and ndcg in metric #45

Closed yinpu closed 1 year ago

yinpu commented 1 year ago

Change the calculation of hit and ndcg in metric

  1. Hit Ratio refer to:https://blog.csdn.net/weixin_31866177/article/details/116741555
  2. NDCG IDCG should be the DCG score of the true label

Please review, I'm not sure if it's correct