关于代码中NDCG@N计算方式的问题

[lazy2panda]

论文中的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。

请问这里应该怎么理解？谢谢

[cenyk1230]

Hi @lazy2panda , 原始代码里的ndcg计算确实跟论文里不一致，可以参见issue #6 。dev分支对ndcg计算进行了修复。

[lazy2panda]

好的，感谢~