使用sklearn计算NDCG时出错

Getting error while calculating NDCG using sklearn

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