python - sklearn导入错误: cannot import name plot_roc_curve

Question

我正在尝试按照 example 绘制具有交叉验证的接收者操作特征 (ROC) 曲线。 sklearn 的文档中提供了。但是，以下导入在 python2 和 python3 中都会出现 ImportError。

from sklearn.metrics import plot_roc_curve

错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: cannot import name plot_roc_curve

python-2.7 sklearn 版本:0.20.2。

python-3.6 sklearn 版本:0.21.3。

我发现以下导入工作正常，但与 plot_roc_curve 不太一样。

from sklearn.metrics import roc_curve

plot_roc_curve 是否已弃用？有人可以尝试一下代码并让我知道 sklearn 版本是否有效吗？

Answer 1

plot_roc_curve 已在 1.2 版本中删除。从 1.2 开始，改用 RocCurveDisplay:

sklearn 1.2 之前:

from sklearn.metrics import plot_roc_curve
svc_disp = plot_roc_curve(svc, X_test, y_test)
rfc_disp = plot_roc_curve(rfc, X_test, y_test, ax=svc_disp.ax_)

来自 sklearn 1.2:

from sklearn.metrics import RocCurveDisplay
svc_disp = RocCurveDisplay.from_estimator(svc, X_test, y_test)
rfc_disp = RocCurveDisplay.from_estimator(rfc, X_test, y_test, ax=svc_disp.ax_)