我试图计算二进制相关性的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
回答:
尝试获取包含排名的指标(参见文档)对于单个真实-预测对来说没有意义(虽然承认这里的错误消息不是很说明问题);你至少需要两个对:
y_true = [[3,3]]
y_score = [[5, 5]]
metrics.ndcg_score(y_true, y_score)
# 0.9999999999999998