使用 sklearn 计算 NDCG 时出错

【问题标题】：Getting error while calculating NDCG using sklearn使用 sklearn 计算 NDCG 时出错
【发布时间】：2021-05-21 06:34:46
【问题描述】：

我正在尝试计算二元相关性的 NDCG 分数：

from sklearn import metrics

y_true = [[3]]
y_score = [[5]]
metrics.ndcg_score(y_true, y_score)

然后出现错误

ValueError: Only ('multilabel-indicator', 'continuous-multioutput', 'multiclass-multioutput') formats are supported. Got binary instead

【问题讨论】：

【解决方案1】：

试图获得包括排名（参见docs）的此类指标对于 single 真实预测对没有任何意义（尽管承认错误消息在这里的信息量不是很大）；你至少需要两对：

y_true = [[3,3]]
y_score = [[5, 5]]
metrics.ndcg_score(y_true, y_score)
# 0.9999999999999998

【讨论】：