如何通过更改熊猫数据框中的参数来绘制ROC曲线

Question

我正在尝试通过更改pandas数据框中一个单元格中的变量在一个绘图上绘制多个ROC曲线。

因此，在特定行中，如果总计超过特定阈值，则它将被分类为发票。我想在不同的总数阈值上绘制不同的曲线。

到目前为止，这是我测量基本指标的代码，它是试图绘制ROC曲线的尝试，但到目前为止，我一直没有成功。

scala> object X { type a }
defined object X

scala> val x: X.a = null.asInstanceOf[X.a]
x: X.a = null

scala> def x: X.a = ???
x: X.a

scala> def x: X.a = null
                    ^
       error: type mismatch;
        found   : Null(null)
        required: X.a

我得到的错误是

fpr [i]，tpr [i]，_ = roc_curve（y_true，probas [:, i]， IndexError：数组的索引过多

任何帮助将不胜感激。

Answer 1

以下文档提到了skplt.metrics.plot_roc_curve针对分类器返回的每个类都采用了地面真实值（正确）目标值和预测概率。因此，您应该更改第二个输入的projection_array。

https://scikit-plot.readthedocs.io/en/stable/metrics.html?highlight=roc#scikitplot.metrics.plot_roc