]jChSSKJr SSKrSSKJr SSKJr SSKJ r SSK J r SSK J r "SS 5rg) )productN) is_classifier)confusion_matrix)check_matplotlib_support)_validate_style_kwargs) unique_labelsc\rSrSrSrSS.SjrSSSSSSSSS .S jr\SSSSSSSSSSSSS . S j5r\SSSSSSSSSSSSS . S j5r Sr g)ConfusionMatrixDisplayaConfusion Matrix visualization. It is recommended to use :func:`~sklearn.metrics.ConfusionMatrixDisplay.from_estimator` or :func:`~sklearn.metrics.ConfusionMatrixDisplay.from_predictions` to create a :class:`ConfusionMatrixDisplay`. All parameters are stored as attributes. For general information regarding `scikit-learn` visualization tools, see the :ref:`Visualization Guide `. For guidance on interpreting these plots, refer to the :ref:`Model Evaluation Guide `. Parameters ---------- confusion_matrix : ndarray of shape (n_classes, n_classes) Confusion matrix. display_labels : ndarray of shape (n_classes,), default=None Display labels for plot. If None, display labels are set from 0 to `n_classes - 1`. 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. See Also -------- confusion_matrix : Compute Confusion Matrix to evaluate the accuracy of a classification. ConfusionMatrixDisplay.from_estimator : Plot the confusion matrix given an estimator, the data, and the label. ConfusionMatrixDisplay.from_predictions : Plot the confusion matrix given the true and predicted labels. Examples -------- >>> import matplotlib.pyplot as plt >>> 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() <...> >>> plt.show() N)display_labelscXlX lg)Nrr )selfrr s c/home/mande/repo/quber/.venv/lib/python3.13/site-packages/sklearn/metrics/_plot/confusion_matrix.py__init__ConfusionMatrixDisplay.__init__Ts 0,Tviridis horizontal)include_valuescmapxticks_rotation values_formataxcolorbarim_kwtext_kwc [S5 SSKJn UcU R5upO URn UR n U R Sn [SUS9n U=(d 0n[X5nU=(d 0nUR"U 40UD6Ul SUl URRS5URRS5pU(Ga[R"U [S9Ul U R!5U R#5-S- n[%['U 5['U 55HunnU UU4U:aUOUnUcW[)U UU4S 5nU R*R,S :wa+[)U UU4S 5n[/U5[/U5:aUnO[)U UU4U5n[S S US 9n[UU5nUR0"UUU40UD6URUU4'M UR2c[R4"U 5nO UR2nU(aU R7URUS9 UR9[R4"U 5[R4"U 5UUSSS9 UR;U S- S45 U R=UR?5US9 Xl XPl!U$)atPlot 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 'd' or '.2g' whichever is shorter. ax : matplotlib axes, default=None Axes object to plot on. If `None`, a new figure and axes is created. colorbar : bool, default=True Whether or not to add a colorbar to the plot. im_kw : dict, default=None Dict with keywords passed to `matplotlib.pyplot.imshow` call. text_kw : dict, default=None Dict with keywords passed to `matplotlib.pyplot.text` call. .. versionadded:: 1.2 Returns ------- display : :class:`~sklearn.metrics.ConfusionMatrixDisplay` Returns a :class:`~sklearn.metrics.ConfusionMatrixDisplay` instance that contains all the information to plot the confusion matrix. zConfusionMatrixDisplay.plotrNnearest) interpolationrg?)dtypeg@z.2gfdcenter)havacolor)rz True labelzPredicted label)xticksyticks xticklabels yticklabelsylabelxlabelg?g)rotation)"rmatplotlib.pyplotpyplotsubplotsfigurershapedictrimshowim_text_rnp empty_likeobjectmaxminrrangeformatr!kindlentextr arangersetset_ylimsetpget_xticklabelsfigure_ax_)rrrrrrrrrpltfigcm n_classes default_im_kwcmap_mincmap_maxthreshijr'text_cmtext_ddefault_text_kwargs text_kwargsr s rplotConfusionMatrixDisplay.plotXsmf !!>?' :llnGC))C  " "HHQK 94@  &}<-R99R)5) !XX]]1-txx}}S/A( r8DJffh)S0Fi 0% 2BC1$&q!tHv$58 ($R1Xu5Gxx}}+!'1a4#!6v;W5&,G$R1X}=G&*h85&Q#45H'R #%771a#HK#H 1a4 D"    &YYy1N!00N  LLbL ) 99Y'99Y'&&$   Y_d+, ##%@  r) labels sample_weight normalizer rrrrrrrrc URS3n[U5 [U5(d[US35eUR U5nUR UUUUUUUU U U U U UUS9$)a Plot Confusion Matrix given an estimator and some data. For general information regarding `scikit-learn` visualization tools, see the :ref:`Visualization Guide `. For guidance on interpreting these plots, refer to the :ref:`Model Evaluation Guide `. .. versionadded:: 1.0 Parameters ---------- estimator : estimator instance Fitted classifier or a fitted :class:`~sklearn.pipeline.Pipeline` in which the last estimator is a classifier. X : {array-like, sparse matrix} of shape (n_samples, n_features) Input values. y : array-like of shape (n_samples,) Target values. labels : array-like of shape (n_classes,), default=None List of labels to index the confusion matrix. This may be used to reorder or select a subset of labels. If `None` is given, those that appear at least once in `y_true` or `y_pred` are used in sorted order. sample_weight : array-like of shape (n_samples,), default=None Sample weights. normalize : {'true', 'pred', 'all'}, default=None Either to normalize the counts display in the matrix: - if `'true'`, the confusion matrix is normalized over the true conditions (e.g. rows); - if `'pred'`, the confusion matrix is normalized over the predicted conditions (e.g. columns); - if `'all'`, the confusion matrix is normalized by the total number of samples; - if `None` (default), the confusion matrix will not be normalized. display_labels : array-like of shape (n_classes,), default=None Target names used for plotting. By default, `labels` will be used if it is defined, otherwise the unique labels of `y_true` and `y_pred` will be used. include_values : bool, default=True Includes values in confusion matrix. 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 'd' or '.2g' whichever is shorter. cmap : str or matplotlib Colormap, default='viridis' Colormap recognized by matplotlib. ax : matplotlib Axes, default=None Axes object to plot on. If `None`, a new figure and axes is created. colorbar : bool, default=True Whether or not to add a colorbar to the plot. im_kw : dict, default=None Dict with keywords passed to `matplotlib.pyplot.imshow` call. text_kw : dict, default=None Dict with keywords passed to `matplotlib.pyplot.text` call. .. versionadded:: 1.2 Returns ------- display : :class:`~sklearn.metrics.ConfusionMatrixDisplay` See Also -------- ConfusionMatrixDisplay.from_predictions : Plot the confusion matrix given the true and predicted labels. Examples -------- >>> import matplotlib.pyplot as plt >>> from sklearn.datasets import make_classification >>> from sklearn.metrics import 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) >>> ConfusionMatrixDisplay.from_estimator( ... clf, X_test, y_test) <...> >>> plt.show() For a detailed example of using a confusion matrix to evaluate a Support Vector Classifier, please see :ref:`sphx_glr_auto_examples_model_selection_plot_confusion_matrix.py` z.from_estimatorz only supports classifiers) rZrYr[r rrrrrrrr)__name__rr ValueErrorpredictfrom_predictions)cls estimatorXyrYrZr[r rrrrrrrr method_namey_preds rfrom_estimator%ConfusionMatrixDisplay.from_estimators|o6  -Y'' },FGH H""1%## '))+'$  rc [URS35 UcUc [X5nOUn[UUUUUS9nU"XS9nUR UU U UU U U US9$)a Plot Confusion Matrix given true and predicted labels. For general information regarding `scikit-learn` visualization tools, see the :ref:`Visualization Guide `. For guidance on interpreting these plots, refer to the :ref:`Model Evaluation Guide `. .. versionadded:: 1.0 Parameters ---------- y_true : array-like of shape (n_samples,) True labels. y_pred : array-like of shape (n_samples,) The predicted labels given by the method `predict` of an classifier. labels : array-like of shape (n_classes,), default=None List of labels to index the confusion matrix. This may be used to reorder or select a subset of labels. If `None` is given, those that appear at least once in `y_true` or `y_pred` are used in sorted order. sample_weight : array-like of shape (n_samples,), default=None Sample weights. normalize : {'true', 'pred', 'all'}, default=None Either to normalize the counts display in the matrix: - if `'true'`, the confusion matrix is normalized over the true conditions (e.g. rows); - if `'pred'`, the confusion matrix is normalized over the predicted conditions (e.g. columns); - if `'all'`, the confusion matrix is normalized by the total number of samples; - if `None` (default), the confusion matrix will not be normalized. display_labels : array-like of shape (n_classes,), default=None Target names used for plotting. By default, `labels` will be used if it is defined, otherwise the unique labels of `y_true` and `y_pred` will be used. include_values : bool, default=True Includes values in confusion matrix. 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 'd' or '.2g' whichever is shorter. cmap : str or matplotlib Colormap, default='viridis' Colormap recognized by matplotlib. ax : matplotlib Axes, default=None Axes object to plot on. If `None`, a new figure and axes is created. colorbar : bool, default=True Whether or not to add a colorbar to the plot. im_kw : dict, default=None Dict with keywords passed to `matplotlib.pyplot.imshow` call. text_kw : dict, default=None Dict with keywords passed to `matplotlib.pyplot.text` call. .. versionadded:: 1.2 Returns ------- display : :class:`~sklearn.metrics.ConfusionMatrixDisplay` See Also -------- ConfusionMatrixDisplay.from_estimator : Plot the confusion matrix given an estimator, the data, and the label. Examples -------- >>> import matplotlib.pyplot as plt >>> from sklearn.datasets import make_classification >>> from sklearn.metrics import 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) >>> y_pred = clf.predict(X_test) >>> ConfusionMatrixDisplay.from_predictions( ... y_test, y_pred) <...> >>> plt.show() z.from_predictions)rZrYr[r)rrrrrrrr)rr]rrrW)ray_truerfrYrZr[r rrrrrrrrrKdisps rr`'ConfusionMatrixDisplay.from_predictionsasn !CLL>1B!CD  !~!.v!>!'   '  BFyy)+'  r)rHrr rGr6r7) r] __module__ __qualname____firstlineno____doc__rrW classmethodrgr`__static_attributes__rrr r sBH<@- $ rh$  #R R h $  !Q Q rr ) itertoolsrnumpyr8 sklearn.basersklearn.metricsr$sklearn.utils._optional_dependenciesrsklearn.utils._plottingrsklearn.utils.multiclassrr rsrrr{s(&,I:2d d r