confusion_matrix#

EstimatorReport.metrics.confusion_matrix(*, data_source='test')[source]#

Plot the confusion matrix.

The confusion matrix shows the counts of correct and incorrect classifications for each class.

Parameters:
data_source{“test”, “train”}, default=”test”

The data source to use.

  • “test” : use the test set provided when creating the report.

  • “train” : use the train set provided when creating the report.

Returns:
ConfusionMatrixDisplay

The confusion matrix display.

Examples

>>> from sklearn.datasets import load_breast_cancer
>>> from sklearn.linear_model import LogisticRegression
>>> from skore import train_test_split
>>> from skore import EstimatorReport
>>> X, y = load_breast_cancer(return_X_y=True)
>>> split_data = train_test_split(X=X, y=y, random_state=0, as_dict=True)
>>> classifier = LogisticRegression(max_iter=10_000)
>>> report = EstimatorReport(classifier, **split_data)
>>> display = report.metrics.confusion_matrix()
>>> display.plot()

With specific threshold for binary classification:

>>> display = report.metrics.confusion_matrix()
>>> display.plot(threshold_value=0.7)