o
    ?HhN                     @   s   d Z ddlZddlmZ ddlmZ ddlZddlm	Z	 ddl
mZ ddlmZ d	d
lmZmZ d	dlmZmZ d#ddZd#ddZeeedZdd Zdd Zdd Zdd Zd$ddZd#ddZd#dd Zd!d" ZdS )%zAUtilities to handle multiclass/multioutput target in classifiers.    N)Sequence)chain)issparse   )get_namespace)VisibleDeprecationWarning   )attach_uniquecached_unique)_assert_all_finitecheck_arrayc                 C   s8   t | |d\}}t| ds|rt|| |dS t| S )Nxp	__array__)r   hasattrr
   asarraysetyr   is_array_api_compliant r   X/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sklearn/utils/multiclass.py_unique_multiclass   s   r   c                 C   s.   t | |d\}}|t| dg ddjd S )Nr   r   csrcsccoo)
input_nameaccept_sparser   )r   aranger   shape)r   r   _r   r   r   _unique_indicator   s   r"   )binary
multiclassmultilabel-indicatorc                     s2  t | ddi} t|  \}t| dkrtdtdd | D }|ddhkr)dh}t|d	kr5td
| | }|dkrNttdd | D d	krNtdt|d  s^tdt|  |rr	 fdd| D }
|S tt fdd| D }ttdd |D d	krtdt|S )a  Extract an ordered array of unique labels.

    We don't allow:
        - mix of multilabel and multiclass (single label) targets
        - mix of label indicator matrix and anything else,
          because there are no explicit labels)
        - mix of label indicator matrices of different sizes
        - mix of string and integer labels

    At the moment, we also don't allow "multiclass-multioutput" input type.

    Parameters
    ----------
    *ys : array-likes
        Label values.

    Returns
    -------
    out : ndarray of shape (n_unique_labels,)
        An ordered array of unique labels.

    Examples
    --------
    >>> from sklearn.utils.multiclass import unique_labels
    >>> unique_labels([3, 5, 5, 5, 7, 7])
    array([3, 5, 7])
    >>> unique_labels([1, 2, 3, 4], [2, 2, 3, 4])
    array([1, 2, 3, 4])
    >>> unique_labels([1, 2, 10], [5, 11])
    array([ 1,  2,  5, 10, 11])
    return_tupleTr   zNo argument has been passed.c                 s   s    | ]}t |V  qd S N)type_of_target).0xr   r   r   	<genexpr>O   s    z unique_labels.<locals>.<genexpr>r#   r$   r   z'Mix type of y not allowed, got types %sr%   c                 s   s&    | ]}t |g d djd V  qdS )r   )r   r   N)r   r    r)   r   r   r   r   r+   \   s    
zCMulti-label binary indicator input with different numbers of labelsNzUnknown label type: %sc                    s   g | ]} |d qS )r   r   r,   _unique_labelsr   r   r   
<listcomp>m   s    z!unique_labels.<locals>.<listcomp>c                 3   s&    | ]}d d  |dD V  qdS )c                 s   s    | ]}|V  qd S r'   r   )r)   ir   r   r   r+   q   s    z*unique_labels.<locals>.<genexpr>.<genexpr>r   Nr   r,   r-   r   r   r+   q   s   $ c                 s   s    | ]}t |tV  qd S r'   )
isinstancestr)r)   labelr   r   r   r+   t   s    z,Mix of label input types (string and number))r	   r   len
ValueErrorr   pop_FN_UNIQUE_LABELSgetreprconcatunique_valuesr   from_iterabler   sorted)ysr   ys_types
label_type	unique_ys	ys_labelsr   r-   r   unique_labels)   sB    
rC   c              	   C   s@   t | \}}|| jdot|||| |j| j| kS )Nreal floating)r   isdtypedtypeboolallastypeint64r   r   r   r   _is_integral_floatz   s    rK   c                 C   s  t | \}}t| dst| ts|ritddddddd}t A tdt zt	| fddi|} W n& tt
fyY } zt|d	rE t	| fdti|} W Y d}~nd}~ww W d   n1 sdw   Y  t| d
rz| jdkrz| jd dks|dS t| r| jdv r|  } || j}t| jdkp|jdks|jdkod|v o| jjdv pt|S t| |d}|jd dk o|| jdpt|S )a~  Check if ``y`` is in a multilabel format.

    Parameters
    ----------
    y : ndarray of shape (n_samples,)
        Target values.

    Returns
    -------
    out : bool
        Return ``True``, if ``y`` is in a multilabel format, else ```False``.

    Examples
    --------
    >>> import numpy as np
    >>> from sklearn.utils.multiclass import is_multilabel
    >>> is_multilabel([0, 1, 0, 1])
    False
    >>> is_multilabel([[1], [0, 2], []])
    False
    >>> is_multilabel(np.array([[1, 0], [0, 0]]))
    True
    >>> is_multilabel(np.array([[1], [0], [0]]))
    False
    >>> is_multilabel(np.array([[1, 0, 0]]))
    True
    r   TFr   r   allow_ndensure_all_finite	ensure_2densure_min_samplesensure_min_featureserrorrF   NComplex data not supportedr    r   r   )doklilbiur      )rG   zsigned integerzunsigned integer)r   r   r1   r   dictwarningscatch_warningssimplefilterr   r   r5   r2   
startswithobjectndimr    r   formattocsrr;   datar4   sizerF   kindrK   r
   rE   )r   r   r   check_y_kwargselabelsr   r   r   is_multilabel   sJ   
 "
rg   c                 C   s(   t | dd}|dvrtd| ddS )aA  Ensure that target y is of a non-regression type.

    Only the following target types (as defined in type_of_target) are allowed:
        'binary', 'multiclass', 'multiclass-multioutput',
        'multilabel-indicator', 'multilabel-sequences'

    Parameters
    ----------
    y : array-like
        Target values.
    r   r   )r#   r$   zmulticlass-multioutputr%   zmultilabel-sequenceszUnknown label type: zy. Maybe you are trying to fit a classifier, which expects discrete classes on a regression target with continuous values.N)r(   r5   )r   y_typer   r   r   check_classification_targets   s   
rj    Fc                    s  t \}} fdd}ttststdr"tt p#|}|s,td jjdv }|r8tdt	r>dS t
ddd	d	d
d
d}t E tdt tsztfddi|W n& ttfy }	 zt|	dru tfdti|W Y d}	~	nd}	~	ww W d   n1 sw   Y  z1trd
gddf nd
 }
t|
trtdt t|
dst|
trt|
tstdW n	 ty   Y nw jdvr| S tjsjdkrdS | S tsjtkrtjd
 ts| S jdkrjd dkrd}nd}|jdr:tr"jn}||||tkr:t | d d| S t|
rB|
j}
t!jd
 dksYjdkr]t"|
dkr]d| S dS )a
  Determine the type of data indicated by the target.

    Note that this type is the most specific type that can be inferred.
    For example:

    * ``binary`` is more specific but compatible with ``multiclass``.
    * ``multiclass`` of integers is more specific but compatible with ``continuous``.
    * ``multilabel-indicator`` is more specific but compatible with
      ``multiclass-multioutput``.

    Parameters
    ----------
    y : {array-like, sparse matrix}
        Target values. If a sparse matrix, `y` is expected to be a
        CSR/CSC matrix.

    input_name : str, default=""
        The data name used to construct the error message.

        .. versionadded:: 1.1.0

    raise_unknown : bool, default=False
        If `True`, raise an error when the type of target returned by
        :func:`~sklearn.utils.multiclass.type_of_target` is `"unknown"`.

        .. versionadded:: 1.6

    Returns
    -------
    target_type : str
        One of:

        * 'continuous': `y` is an array-like of floats that are not all
          integers, and is 1d or a column vector.
        * 'continuous-multioutput': `y` is a 2d array of floats that are
          not all integers, and both dimensions are of size > 1.
        * 'binary': `y` contains <= 2 discrete values and is 1d or a column
          vector.
        * 'multiclass': `y` contains more than two discrete values, is not a
          sequence of sequences, and is 1d or a column vector.
        * 'multiclass-multioutput': `y` is a 2d array that contains more
          than two discrete values, is not a sequence of sequences, and both
          dimensions are of size > 1.
        * 'multilabel-indicator': `y` is a label indicator matrix, an array
          of two dimensions with at least two columns, and at most 2 unique
          values.
        * 'unknown': `y` is array-like but none of the above, such as a 3d
          array, sequence of sequences, or an array of non-sequence objects.

    Examples
    --------
    >>> from sklearn.utils.multiclass import type_of_target
    >>> import numpy as np
    >>> type_of_target([0.1, 0.6])
    'continuous'
    >>> type_of_target([1, -1, -1, 1])
    'binary'
    >>> type_of_target(['a', 'b', 'a'])
    'binary'
    >>> type_of_target([1.0, 2.0])
    'binary'
    >>> type_of_target([1, 0, 2])
    'multiclass'
    >>> type_of_target([1.0, 0.0, 3.0])
    'multiclass'
    >>> type_of_target(['a', 'b', 'c'])
    'multiclass'
    >>> type_of_target(np.array([[1, 2], [3, 1]]))
    'multiclass-multioutput'
    >>> type_of_target([[1, 2]])
    'multilabel-indicator'
    >>> type_of_target(np.array([[1.5, 2.0], [3.0, 1.6]]))
    'continuous-multioutput'
    >>> type_of_target(np.array([[0, 1], [1, 1]]))
    'multilabel-indicator'
    c                     s(   r r nd} t d|  ddS )zdDepending on the value of raise_unknown, either raise an error or return
        'unknown'.
        ra   zUnknown label type for z: unknown)r5   )inputr   raise_unknownr   r   r   _raise_or_return4  s   z(type_of_target.<locals>._raise_or_returnr   z:Expected array-like (array or non-string sequence), got %r)SparseSeriesSparseArrayz1y cannot be class 'SparseSeries' or 'SparseArray'r%   TFr   rL   rR   rF   NrS   zSupport for labels represented as bytes is deprecated in v1.5 and will error in v1.7. Convert the labels to a string or integer format.zYou appear to be using a legacy multi-label data representation. Sequence of sequences are no longer supported; use a binary array or sparse matrix instead - the MultiLabelBinarizer transformer can convert to this format.)r   r   r   r#   r   z-multioutputrk   rD   rh   
continuousr$   )#r   r1   r   r   r   r2   r5   	__class____name__rg   rX   rY   rZ   r[   r   r   r\   r]   byteswarnFutureWarning
IndexErrorr^   minr    rF   flatrE   ra   anyrI   intr   r
   r4   )r   r   ro   r   r   rp   validsparse_pandasrd   re   first_row_or_valsuffixra   r   rn   r   r(      s   M

	 "




"
.r(   c                 C   sr   t | dddu r|du rtd|dur7t | dddur0t| jt|s.td|| jf dS t|| _dS dS )a"  Private helper function for factorizing common classes param logic.

    Estimators that implement the ``partial_fit`` API need to be provided with
    the list of possible classes at the first call to partial_fit.

    Subsequent calls to partial_fit should check that ``classes`` is still
    consistent with a previous value of ``clf.classes_`` when provided.

    This function returns True if it detects that this was the first call to
    ``partial_fit`` on ``clf``. In that case the ``classes_`` attribute is also
    set on ``clf``.

    classes_Nz8classes must be passed on the first call to partial_fit.zD`classes=%r` is not the same as on last call to partial_fit, was: %rTF)getattrr5   nparray_equalr   rC   )clfclassesr   r   r   _check_partial_fit_first_call  s   
r   c                 C   s  g }g }g }| j \}}|durt|}t| r|  } t| j}t|D ]}| j| j| | j|d   }	|durJ||	 }
t	|t	|
 }nd}
| j d ||  }tj
| j| j| | j|d   dd\}}tj||
d}d|v r||dk  |7  < d|vr|| | j d k rt|dd}t|d|}|| ||j d  |||	   q&n3t|D ].}tj
| dd|f dd\}}|| ||j d  tj||d}|||	   q|||fS )az  Compute class priors from multioutput-multiclass target data.

    Parameters
    ----------
    y : {array-like, sparse matrix} of size (n_samples, n_outputs)
        The labels for each example.

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

    Returns
    -------
    classes : list of size n_outputs of ndarray of size (n_classes,)
        List of classes for each column.

    n_classes : list of int of size n_outputs
        Number of classes in each column.

    class_prior : list of size n_outputs of ndarray of size (n_classes,)
        Class distribution of each column.
    Nr   r   T)return_inverse)weights)r    r   r   r   tocscdiffindptrrangeindicessumuniquera   bincountinsertappend)r   sample_weightr   	n_classesclass_prior	n_samples	n_outputsy_nnzkcol_nonzeronz_samp_weightzeros_samp_weight_sum	classes_ky_kclass_prior_kr   r   r   class_distribution  sH   





r   c           
      C   s  | j d }t||f}t||f}d}t|D ]X}t|d |D ]N}|dd|f  |dd|f 8  < |dd|f  |dd|f 7  < || dd|f dk|f  d7  < || dd|f dk|f  d7  < |d7 }q"q|dt|d   }	||	 S )ay  Compute a continuous, tie-breaking OvR decision function from OvO.

    It is important to include a continuous value, not only votes,
    to make computing AUC or calibration meaningful.

    Parameters
    ----------
    predictions : array-like of shape (n_samples, n_classifiers)
        Predicted classes for each binary classifier.

    confidences : array-like of shape (n_samples, n_classifiers)
        Decision functions or predicted probabilities for positive class
        for each binary classifier.

    n_classes : int
        Number of classes. n_classifiers must be
        ``n_classes * (n_classes - 1 ) / 2``.
    r   r   NrW   )r    r   zerosr   abs)
predictionsconfidencesr   r   votessum_of_confidencesr   r0   jtransformed_confidencesr   r   r   _ovr_decision_function  s    
$$$$
r   r'   )rk   F)__doc__rY   collections.abcr   	itertoolsr   numpyr   scipy.sparser   utils._array_apir   utils.fixesr   _uniquer	   r
   
validationr   r   r   r"   r7   rC   rK   rg   rj   r(   r   r   r   r   r   r   r   <module>   s2    

QI
 
J
#J