sklearn.metrics
.ConfusionMatrixDisplay¶
- class sklearn.metrics.ConfusionMatrixDisplay¶
Visualización de la matriz de confusión.
Se recomienda utilizar
plot_confusion_matrix
para crear unaConfusionMatrixDisplay
. Todos los parámetros se almacenan como atributos.Más información en el Manual de Usuario.
- Parámetros
- confusion_matrixndarray de forma (n_samples, n_classes)
Matriz de confusión.
- display_labelsndarray de forma (n_classes,), default=None
Las etiquetas de visualización para la gráfica. Si es None, las etiquetas de visualización se establecen de 0 a
n_classes - 1
.
- Atributos
- im_AxesImage de matplotlib
Imagen que representa la matriz de confusión.
- text_ndarray de forma (n_classes, n_classes), dtype=matplotlib Text, o None
Arreglo de ejes de matplotlib.
None
siinclude_values
es falso.- ax_Ejes de matplotlib
Ejes con matriz de confusión.
- figure_Figura de matplotlib
Figura que contiene la matriz de confusión.
Ver también
confusion_matrix
Calcula la matriz de confusión para evaluar la precisión de una clasificación.
plot_confusion_matrix
Gráfica de la Matriz de Confusión.
Ejemplos
>>> from sklearn.datasets import make_classification >>> from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay >>> from sklearn.model_selection import train_test_split >>> from sklearn.svm import SVC >>> X, y = make_classification(random_state=0) >>> X_train, X_test, y_train, y_test = train_test_split(X, y, ... random_state=0) >>> clf = SVC(random_state=0) >>> clf.fit(X_train, y_train) SVC(random_state=0) >>> predictions = clf.predict(X_test) >>> cm = confusion_matrix(y_test, predictions, labels=clf.classes_) >>> disp = ConfusionMatrixDisplay(confusion_matrix=cm, ... display_labels=clf.classes_) >>> disp.plot()
Métodos
Visualización de la gráfica.
- plot()¶
Visualización de la gráfica.
- Parámetros
- include_valuesbooleano, default=True
Incluye valores en la matriz de confusión.
- cmapcadena de caracteres o matplotlib Colormap, default=”viridis”
Mapa de colores reconocido por matplotlib.
- xticks_rotation{“vertical”, “horizontal”} o flotante, default=”horizontal”
Rotación de etiquetas xtick.
- values_formatcadena de caracteres, default=None
Especificación de formato para los valores en la matriz de confusión. Si en
None
, la especificación de formato es “d” o “.2g” que sea más corta.- axejes matplotlib, default=None
Objeto de ejes en los que dibujar. Si es
None
, se crea una nueva figura y ejes.- colorbarbooleano, default=True
Añadir o no una barra de colores al gráfico.
- Devuelve
- display
ConfusionMatrixDisplay
- display