如何使用sklearn计算具有二进制相关性的NDCG？

问题描述

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

from sklearn.metrics import ndcg_score
y_true = [0,1,0]
y_pred = [0,0]
ndcg_score(y_true,y_pred)

并得到：

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

有没有办法使这项工作成功？

解决方法

请尝试：

from sklearn.metrics import ndcg_score
y_true = [[0,1,0]]
y_pred = [[0,0]]
ndcg_score(y_true,y_pred)
1.0

请注意docs中的预期形状：

y_true：ndarray，形状（n_samples，n_labels）
y_score：ndarray，形状（n_samples，n_labels）