Coder-Yu / QRec

QRec: A Python Framework for quick implementation of recommender systems (TensorFlow Based)
1.59k stars 406 forks source link

NDCG计算 #258

Closed laowangzi closed 1 year ago

laowangzi commented 1 year ago

作者你好, 首先感谢你们的开源精神,为社区贡献了如此完整和优秀的代码。 在学习过程中,我对util.measure中你们对NDCG指标的实现部分存在疑问。在计算DCG时,分母部分你们使用math.log(n+2)实现,默认底数为e,而在DCG的公式和大部分其他工作的实现中,底数一般为2。我认为这个差异会造成NDCG数值计算结果的不同,在交叉对比其他工作时可能会造成一定影响。个人拙见,不知道你们如何看待这个问题?

Coder-Yu commented 1 year ago

log的底数对计算结果没有影响,你可以自己推导一下