Python，ROC曲线

Question

我使用了 sklearn.metrics 中的 roc_curve。我看到这个错误:

我有二元分类器的结果 (true_labels ={0,1})。我需要绘制 ROC 曲线。所以我写道:

X, Y, thresholds = sklearn.metrics.roc_curve(true_labels, prediction_scores, pos_label=1)

我看到这个错误:

raise ValueError("{0} format is not supported".format(y_type))

ValueError: multilabel-indicator format is not supported

如有任何帮助，我们将不胜感激!

Answer 1

根据评论中的讨论，您的问题似乎是数组上的额外维度。您可以通过在将输入数组传递到 roc_curve 之前对输入数组应用 numpy.squeeze 来删除单一维度。如果您知道这只是额外的单一维度，您可以使用 ravel 方法将数组视为一维数组。