0Ph@pddlmZddlZddlmZddlmZddlm Z ddl m Z dd l m Z Gd d ZdS) )productN) is_classifier)check_matplotlib_support)_validate_style_kwargs) unique_labels)confusion_matrixceZdZdZdddZdddddddddd Zeddddddddddddd d Zeddddddddddddd d ZdS) ConfusionMatrixDisplaya Confusion Matrix visualization. It is recommend 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. Read more in the :ref:`User 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_labelsc"||_||_dS)Nr r )selfr r s f/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/sklearn/metrics/_plot/confusion_matrix.py__init__zConfusionMatrixDisplay.__init__Qs 0,Tviridis horizontal)include_valuescmapxticks_rotation values_formataxcolorbarim_kwtext_kwctdddlm} || \} }n|j} |j} | jd} td|} |pi}t| |}|pi}|j | fi||_ d|_ |j d|j d}}|rGtj| t|_ | | zdz }t%t'| t'| D]\}}| ||f|kr|n|}|ct)| ||fd }| jjd kr:t)| ||fd }t/|t/|kr|}nt)| ||f|}td d | }t||}|j|||fi||j ||f<|jtj| }n|j}|r| |j ||tj| tj| ||dd|| dz df| ||| |_ ||_!|S)aLPlot 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.pyplotpyplotsubplotsfigurer shapedictrimshowim_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 rplotzConfusionMatrixDisplay.plotUsf !!>???'''''' :llnnGC)C  "HQK 94@@@  &}e<<-R29R))5)) !X]]1--tx}}S/A/A(  Ir888DJffhh)S0Fi 0 0% 2B2BCC I I1$&q!tHv$5$58 ($R1Xu55Gx}++!'1a4#!6!6v;;W55&,G$R1X}==G&*h85&Q&Q&Q#45H'RR #*271a#H#HK#H#H 1a4    &Yy11NN!0N  * LLbL ) ) ) 9Y''9Y''&&$     Y_d+,,, ##%%@@@  r) labels sample_weight normalizer rrrrrrrrc |jd}t|t|st|d||}||||||||| | | | | ||S)aPlot Confusion Matrix given an estimator and some data. Read more in the :ref:`User 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() z.from_estimatorz only supports classifiers) rYrXrZr rrrrrrrr)__name__rr ValueErrorpredictfrom_predictions)cls estimatorXyrXrYrZr rrrrrrrr method_namey_preds rfrom_estimatorz%ConfusionMatrixDisplay.from_estimatorsn666  ---Y'' I GGGHH H""1%%## '))+'$   rc t|jd||t||}n|}t|||||}|||}||| | || | | |S)aWPlot Confusion Matrix given true and predicted labels. Read more in the :ref:`User 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_predictionsN)rYrXrZr)rrrrrrrr)rr\rr rW)r`y_truererXrYrZr rrrrrrrrrKdisps rr_z'ConfusionMatrixDisplay.from_predictionsWsh !CL!C!C!CDDD  !~!.vv!>!>!'   '    sB~FFFyy)+'   r) r\ __module__ __qualname____doc__rrW classmethodrfr_rrr r s??B<@----- $ rrrrrh$  #K K K K [K Z $  !N N N N [N N N rr ) itertoolsrnumpyr8baserutils._optional_dependenciesrutils._plottingrutils.multiclassrr r rnrrrvs!!!!!!DDDDDD555555------W W W W W W W W W W r