机器学习

用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官方文档 (讲得没本文详细) 本文链接