web-dev-qa-db-fra.com

sklearn ImportError: impossible d'importer le nom plot_roc_curve

J'essaie de tracer une courbe des caractéristiques de fonctionnement du récepteur (ROC) avec validation croisée, en suivant le exemple fourni dans la documentation de sklearn. Cependant, l'importation suivante donne un ImportError, dans les deux python2 et python3.

from sklearn.metrics import plot_roc_curve

Erreur:

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

python-2.7 version sklearn: 0.20.2.

python-3.6 version sklearn: 0.21.3.

J'ai trouvé que l'importation suivante fonctionne bien, mais ce n'est pas tout à fait la même chose que plot_roc_curve.

from sklearn.metrics import roc_curve

Est plot_roc_curve obsolète? Quelqu'un pourrait-il essayer le code et me faire savoir la version de sklearn si cela fonctionne?

4
Achintha Ihalage

Installez scikit-plot et importez la métrique à partir de là:

from scikitplot.metrics import plot_roc_curve
0
Talha Rifaai