NDCG指标讲解

发布时间 2023-09-04 16:48:20作者: ANewPro

NDCG

NDCG是Normalized Discount Cumulative Gain

NDCG可以在搜索和推荐任务中衡量返回的Item列表是否优秀

G-CG-DCG-NDCG

G代表Gain

表示一个列表中的Item的相关性分数,rel(i)表示item(i)相关性得

Gain=rel(i)

CG就是Cumulative Gain表示对K个item的Gain进行累加

\(CG_k=\sum_{i=1}^krel(i)\)

DCG就是Discount Cumulative Gain 考虑排序的因为,使得排名靠前的item增益更高,对排名靠后的item进行折损

\(DCG_k=\sum^k_{i=1}\frac{rel(i)}{log_2(i+1)}\)

相当于给每个item的相关性打个折扣,item越靠后,折扣越大

IDCG 就是理想的DCG,是根据rel(i)进行降序排列,即排列到最好状态,算出最好排列的DCG,就是IDCG

NDCG就是归一化的DCG

\(NDCG=\frac{DCG}{IDCG}\)