sklearn.metrics
.plot_roc_curve¶
- sklearn.metrics.plot_roc_curve()¶
Graficar la curva de características operativas del receptor (ROC).
Los argumentos de palabras clave adicionales se pasarán al
plot
de matplotlib.Más información en el Manual de Usuario.
- Parámetros
- estimatorinstancia del estimador
Clasificador ajustado o un
Pipeline
ajustado en el que el último estimador es un clasificador.- X{array-like, sparse matrix} de forma (n_samples, n_features)
Valores de entrada
- yarray-like de forma (n_samples,)
Valores objetivo.
- sample_weightarray-like de forma (n_samples,), default=None
Ponderaciones de muestra.
- drop_intermediatebooleano, default=True
Si se caen algunos umbrales subóptimos que no aparecen en una curva ROC trazada. Esto es útil para crear curvas ROC más ligeras.
- response_method{“predict_proba”, “decision_function”, “auto”} default=”auto”
Especifica si usar predict_proba o decision_function como respuesta al objetivo. Si se establece en “auto”, predict_proba es probado primero y si no existe decision_function es probado a continuación.
- namecadena de caracteres, default=None
Nombre de la curva ROC para etiquetar. Si es
None
, usa el nombre del estimador.- axejes matplotlib, default=None
Objeto de ejes para graficar. Si es
None
, se crea una nueva figura y ejes.- pos_labelcadena de caracteres o entero, default=None
La clase considerada como la clase positiva al calcular las métricas de roc auc. Por defecto,
estimators.classes_[1]
es considerada como la clase positiva.Nuevo en la versión 0.24.
- Devuelve
- display
RocCurveDisplay
Objeto que almacena valores calculados.
- display
Ver también
roc_curve
Calcula la curva de características operativas del receptor (ROC).
RocCurveDisplay
Visualización de curvas ROC.
roc_auc_score
Calcula el área bajo la curva ROC.
Ejemplos
>>> import matplotlib.pyplot as plt >>> from sklearn import datasets, metrics, model_selection, svm >>> X, y = datasets.make_classification(random_state=0) >>> X_train, X_test, y_train, y_test = model_selection.train_test_split( ... X, y, random_state=0) >>> clf = svm.SVC(random_state=0) >>> clf.fit(X_train, y_train) SVC(random_state=0) >>> metrics.plot_roc_curve(clf, X_test, y_test) >>> plt.show()