
    _nd*                     X    d dl ZddlmZ ddlmZ ddlmZ ddlm	Z	  G d	 d
          Z
dS )    N   )_get_response   )	det_curve)_check_pos_label_consistency   )check_matplotlib_supportc                   t    e Zd ZdZddddZeddddddd            Zedddddd	            Zddd
dZdS )DetCurveDisplayaw  DET curve visualization.

    It is recommend to use :func:`~sklearn.metrics.DetCurveDisplay.from_estimator`
    or :func:`~sklearn.metrics.DetCurveDisplay.from_predictions` to create a
    visualizer. All parameters are stored as attributes.

    Read more in the :ref:`User Guide <visualizations>`.

    .. versionadded:: 0.24

    Parameters
    ----------
    fpr : ndarray
        False positive rate.

    fnr : ndarray
        False negative rate.

    estimator_name : str, default=None
        Name of estimator. If None, the estimator name is not shown.

    pos_label : str or int, default=None
        The label of the positive class.

    Attributes
    ----------
    line_ : matplotlib Artist
        DET Curve.

    ax_ : matplotlib Axes
        Axes with DET Curve.

    figure_ : matplotlib Figure
        Figure containing the curve.

    See Also
    --------
    det_curve : Compute error rates for different probability thresholds.
    DetCurveDisplay.from_estimator : Plot DET curve given an estimator and
        some data.
    DetCurveDisplay.from_predictions : Plot DET curve given the true and
        predicted labels.

    Examples
    --------
    >>> import matplotlib.pyplot as plt
    >>> from sklearn.datasets import make_classification
    >>> from sklearn.metrics import det_curve, DetCurveDisplay
    >>> from sklearn.model_selection import train_test_split
    >>> from sklearn.svm import SVC
    >>> X, y = make_classification(n_samples=1000, random_state=0)
    >>> X_train, X_test, y_train, y_test = train_test_split(
    ...     X, y, test_size=0.4, random_state=0)
    >>> clf = SVC(random_state=0).fit(X_train, y_train)
    >>> y_pred = clf.decision_function(X_test)
    >>> fpr, fnr, _ = det_curve(y_test, y_pred)
    >>> display = DetCurveDisplay(
    ...     fpr=fpr, fnr=fnr, estimator_name="SVC"
    ... )
    >>> display.plot()
    <...>
    >>> plt.show()
    N)estimator_name	pos_labelc                >    || _         || _        || _        || _        d S Nfprfnrr   r   )selfr   r   r   r   s        ?lib/python3.11/site-packages/sklearn/metrics/_plot/det_curve.py__init__zDetCurveDisplay.__init__L   s#    ,"    auto)sample_weightresponse_methodr   nameaxc          
          t          | j         d           ||j        j        n|}t          ||||          \  }
} | j        d||
||||d|	S )a\
  Plot DET curve given an estimator and data.

        Read more in the :ref:`User Guide <visualizations>`.

        .. 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.

        sample_weight : array-like of shape (n_samples,), default=None
            Sample weights.

        response_method : {'predict_proba', 'decision_function', 'auto'}                 default='auto'
            Specifies whether to use :term:`predict_proba` or
            :term:`decision_function` as the predicted target response. If set
            to 'auto', :term:`predict_proba` is tried first and if it does not
            exist :term:`decision_function` is tried next.

        pos_label : str or int, default=None
            The label of the positive class. When `pos_label=None`, if `y_true`
            is in {-1, 1} or {0, 1}, `pos_label` is set to 1, otherwise an
            error will be raised.

        name : str, default=None
            Name of DET curve for labeling. If `None`, use the name of the
            estimator.

        ax : matplotlib axes, default=None
            Axes object to plot on. If `None`, a new figure and axes is
            created.

        **kwargs : dict
            Additional keywords arguments passed to matplotlib `plot` function.

        Returns
        -------
        display : :class:`~sklearn.metrics.DetCurveDisplay`
            Object that stores computed values.

        See Also
        --------
        det_curve : Compute error rates for different probability thresholds.
        DetCurveDisplay.from_predictions : Plot DET curve given the true and
            predicted labels.

        Examples
        --------
        >>> import matplotlib.pyplot as plt
        >>> from sklearn.datasets import make_classification
        >>> from sklearn.metrics import DetCurveDisplay
        >>> from sklearn.model_selection import train_test_split
        >>> from sklearn.svm import SVC
        >>> X, y = make_classification(n_samples=1000, random_state=0)
        >>> X_train, X_test, y_train, y_test = train_test_split(
        ...     X, y, test_size=0.4, random_state=0)
        >>> clf = SVC(random_state=0).fit(X_train, y_train)
        >>> DetCurveDisplay.from_estimator(
        ...    clf, X_test, y_test)
        <...>
        >>> plt.show()
        z.from_estimatorN)r   )y_truey_predr   r   r   r    )r	   __name__	__class__r   from_predictions)cls	estimatorXyr   r   r   r   r   kwargsr   s              r   from_estimatorzDetCurveDisplay.from_estimatorR   s    j 	!CL!A!A!ABBB/3|y"++)	
 
 
	 $s# 
'
 
 
 
 	
r   )r   r   r   r   c                    t          | j         d           t          ||||          \  }}	}
t          ||          }|dn|}t	          ||	||          } |j        d||d|S )a	  Plot the DET curve given the true and predicted labels.

        Read more in the :ref:`User Guide <visualizations>`.

        .. versionadded:: 1.0

        Parameters
        ----------
        y_true : array-like of shape (n_samples,)
            True labels.

        y_pred : array-like of shape (n_samples,)
            Target scores, can either be probability estimates of the positive
            class, confidence values, or non-thresholded measure of decisions
            (as returned by `decision_function` on some classifiers).

        sample_weight : array-like of shape (n_samples,), default=None
            Sample weights.

        pos_label : str or int, default=None
            The label of the positive class. When `pos_label=None`, if `y_true`
            is in {-1, 1} or {0, 1}, `pos_label` is set to 1, otherwise an
            error will be raised.

        name : str, default=None
            Name of DET curve for labeling. If `None`, name will be set to
            `"Classifier"`.

        ax : matplotlib axes, default=None
            Axes object to plot on. If `None`, a new figure and axes is
            created.

        **kwargs : dict
            Additional keywords arguments passed to matplotlib `plot` function.

        Returns
        -------
        display : :class:`~sklearn.metrics.DetCurveDisplay`
            Object that stores computed values.

        See Also
        --------
        det_curve : Compute error rates for different probability thresholds.
        DetCurveDisplay.from_estimator : Plot DET curve given an estimator and
            some data.

        Examples
        --------
        >>> import matplotlib.pyplot as plt
        >>> from sklearn.datasets import make_classification
        >>> from sklearn.metrics import DetCurveDisplay
        >>> from sklearn.model_selection import train_test_split
        >>> from sklearn.svm import SVC
        >>> X, y = make_classification(n_samples=1000, random_state=0)
        >>> X_train, X_test, y_train, y_test = train_test_split(
        ...     X, y, test_size=0.4, random_state=0)
        >>> clf = SVC(random_state=0).fit(X_train, y_train)
        >>> y_pred = clf.decision_function(X_test)
        >>> DetCurveDisplay.from_predictions(
        ...    y_test, y_pred)
        <...>
        >>> plt.show()
        z.from_predictions)r   r   N
Classifierr   )r   r   r   )r	   r    r   r   r   plot)r#   r   r   r   r   r   r   r'   r   r   _vizs               r   r"   z DetCurveDisplay.from_predictions   s    V 	!CL!C!C!CDDD'	
 
 
S! 1FCC	#|||	
 
 
 sx32D33F333r   )r   c                   t          d           || j        n|}|i nd|i} |j        di | ddlm} ||                                \  }} |j        t          j        j	        
                    | j                  t          j        j	        
                    | j                  fi |\  | _        | j        d| j         dnd}d|z   }d	|z   }	|                    ||	
           d|v r|                    d           g d}
t          j        j	        
                    |
          }d |
D             }|                    |           |                    |           |                    dd           |                    |           |                    |           |                    dd           || _        |j        | _        | S )au  Plot visualization.

        Parameters
        ----------
        ax : matplotlib axes, default=None
            Axes object to plot on. If `None`, a new figure and axes is
            created.

        name : str, default=None
            Name of DET curve for labeling. If `None`, use `estimator_name` if
            it is not `None`, otherwise no labeling is shown.

        **kwargs : dict
            Additional keywords arguments passed to matplotlib `plot` function.

        Returns
        -------
        display : :class:`~sklearn.metrics.plot.DetCurveDisplay`
            Object that stores computed values.
        zDetCurveDisplay.plotNlabelr   z (Positive label: ) zFalse Positive RatezFalse Negative Rate)xlabelylabelzlower right)loc)	gMbP?g{Gz?g?g?g      ?g?gffffff?gGz?g+?c                     g | ]C}d |z                                   rd                    |          nd                    |          DS )d   z{:.0%}z{:.1%})
is_integerformat).0ss     r   
<listcomp>z(DetCurveDisplay.plot.<locals>.<listcomp>M  sZ     
 
 
 $'7"6"6"8"8PHOOAhooa>P>P
 
 
r   r   r   )r	   r   updatematplotlib.pyplotpyplotsubplotsr+   spstatsnormppfr   r   line_r   setlegend
set_xticksset_xticklabelsset_xlim
set_yticksset_yticklabelsset_ylimax_figurefigure_)r   r   r   r'   line_kwargspltr,   info_pos_labelr2   r3   tickstick_locationstick_labelss                r   r+   zDetCurveDisplay.plot  s   * 	!!7888&*lt"" Lbbwo$$V$$$'''''':LLNNEArHMdh''HMdh''
 
 
 
 7;n6P22222VX 	 '7&7
fV,,,k!!II-I(((GGG**511
 

 
 
 	n%%%
;'''
B
n%%%
;'''
Byr   r   )	r    
__module____qualname____doc__r   classmethodr(   r"   r+   r   r   r   r   r      s        > >@ 484 # # # # #  g
 g
 g
 g
 [g
R  \4 \4 \4 \4 [\4|?D ? ? ? ? ? ? ?r   r   )scipyrA   baser   r1   r   _baser   utilsr	   r   r   r   r   <module>r_      s                    0 0 0 0 0 0 - - - - - -O O O O O O O O O Or   