sklearn.metrics.ConfusionMatrixDisplay

class sklearn.metrics.ConfusionMatrixDisplay(confusion_matrix, display_labels)[source]

Confusion Matrix visualization.

It is recommend to use plot_confusion_matrix to create a ConfusionMatrixDisplay. All parameters are stored as attributes.

Read more in the User Guide.

Parameters

confusion_matrix : ndarray of shape (n_classes, n_classes)

Confusion matrix.

display_labels : ndarray of shape (n_classes,)

Display labels for plot.

Attributes

im_

(matplotlib AxesImage) Image representing the confusion matrix.

text_

(ndarray of shape (n_classes, n_classes), dtype=matplotlib Text, or None) Array of matplotlib axes. None if include_values is false.

ax_

(matplotlib Axes) Axes with confusion matrix.

figure_

(matplotlib Figure) Figure containing the confusion matrix.

Methods

plot([include_values, cmap, …])

Plot visualization.

__init__(confusion_matrix, display_labels)[source]

Initialize self. See help(type(self)) for accurate signature.

plot(include_values=True, cmap='viridis', xticks_rotation='horizontal', values_format=None, ax=None)[source]

Plot visualization.

Parameters

include_values : bool, default=True

Includes values in confusion matrix.

cmap : str or matplotlib Colormap, default=’viridis’

Colormap recognized by matplotlib.

xticks_rotation : {‘vertical’, ‘horizontal’} or float, default=’horizontal’

Rotation of xtick labels.

values_format : str, default=None

Format specification for values in confusion matrix. If None, the format specification is ‘.2g’.

ax : matplotlib axes, default=None

Axes object to plot on. If None, a new figure and axes is created.

Returns

display : ConfusionMatrixDisplay