使用sklearn计算NDCG时出错

问题描述

我正在尝试计算二元相关性的 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

解决方法

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

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