1. 程式人生 > >NDCG評價指標講解

NDCG評價指標講解

Normalized Discounted Cumulative Gain,即NDCG,常用於作為對rank的評價指標,當我們通過模型得出某些元素的ranking的時候,便可以通過NDCG來測評這個rank的準確度,同樣的演算法還包括MAP,MRR等,這裡只講解NDCG。


我們通過一個例項來說明一下這個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