使用sklearn计算NDCG时出错

Question

我试图计算二元关联的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

Answer 1

试图获得这样的指标，包括单个真实预测对的排名（参见docs），没有任何意义（尽管这里的错误信息不是很有用）；您至少需要两对：

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