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方法将您的数组视为一维数组。