Python, ROC曲线

Python, ROC curve

我使用了 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

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

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