機器學習

用scikit-learn計算NDCG

簡介 NDCG是衡量Ranking quality是重要指標。本文將用實際Python例子演示怎樣計算NDCG。 用scikit計算NDCG例子 注意ndcg_score接收的參數都是list of list。後面解釋為什麼是list of list。 如果評估一個排序請求,用以下例子 如果有多個排序請求,比如日誌里記錄的一天收到的所有排序請求,用以下例子(假設有3個排序請求) 3個排序會產生3個NDCG,最後結果為所有NDCG的平均值,作為這麼多請求的總NDCG返回。 還要注意每個排序請求list裏面元素的數量要一樣,不然ndcg_score函數會報錯。 參考 Scikit-learn官方文檔 (講得沒本文詳細) 本文鏈接