关于代码中NDCG@N计算方式的问题
lazy2panda opened this issue · comments
lazy2panda commented
论文中的NDCG@N如下:
代码中的ndcg计算方式如下:
dcg:https://github.com/THUDM/ComiRec/blob/master/src/train.py#L143
idcg:https://github.com/THUDM/ComiRec/blob/master/src/train.py#L148
论文中的dcg求和项分母是k,是指给用户u推荐的第k个item。
论文中的k是否对应dcg计算代码中的no?
我理解no是用户实际交互项中的编号,并不是给用户推荐的第k个item。
请问这里应该怎么理解?谢谢
Codazzy commented
Hi @lazy2panda , 原始代码里的ndcg计算确实跟论文里不一致,可以参见issue #6 。dev分支对ndcg计算进行了修复。
lazy2panda commented
好的,感谢~