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

Question

我试图计算二进制关联的NDCG分数：

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

并得到：

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

有办法让这件事成功吗？

Answer 1

请尝试：

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

注意文档中的预期形状

y_true: ndarray，shape (n_samples，n_labels) y_score: ndarray，shape (n_samples，n_labels)