NDCG评价指标讲解

Normalized Discounted Cumulative Gain,即NDCG,常用于作为对rank的评价指标,当我们通过模型得出某些元素的ranking的时候,便可以通过NDCG来测评这个rank的准确度,同样的算法还包括MAP,MRR等,这里只讲解NDCG。

NDCG评价指标讲解_第1张图片

我们通过一个实例来说明一下这个NDCG的用法

id 1 2 3 4 5
模型评分 1.2 0.7 0.1 0.2 4.0
按模型评分排序(i) 2 3 5 4 1
r(l) 2 1 0 0 0
按Gain排序(j) 1 2 3 3 3

我们首先通过i来计算出DCG值为2.39278,按照j来计算出iDCG的值为3,6309

最后通过公式1计算出NDCG为0.65

你可能感兴趣的:(数据挖掘)