
    KdQ                        d dl Zd dlZd dlmZ d dlmZ d dlm	Z	m
Z
 d dlmZ d dlmZmZmZmZ d dlmZ ed             Zd Zd	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zej         !                    dg dg df ej"        ddgddgddgddgg           ej"        ddgddgddgddgg          fg          d             Z#ej         !                    dg d          d             Z$d Z%d Z&d Z'd Z(d  Z)d! Z*d" Z+d# Z,d$ Z-d% Z.d& Z/d' Z0d( Z1d) Z2d* Z3d+ Z4ej         !                    d,g d-d.d id/fg d-d0dd gid1f ej5        g d-g d-g          d0did1fg d-d0d2id3f ej5        g d-g d-g          d0dd4gid5fgg d67          d8             Z6d9 Z7d: Z8d; Z9d< Z:d= Z;dJd?Z<d@ Z=dA Z>dB Z?ej         !                    dg dCdDgdEz  f ej"        ddgddgddgddgg          dDdDggdEz  fg          dF             Z@ej         !                    dg dG          dH             ZAej         !                    dg d          dI             ZBdS )K    N)clone)DummyClassifierDummyRegressor)NotFittedError)assert_almost_equalassert_array_almost_equalassert_array_equalignore_warnings)_weighted_percentilec           	         |                      |          }|                     |          }t          j        |          }|j        dk    rt          j        |d          }|j        d         }t          |          }|dk    r|g}|g}t          |          D ]}||         j        d         |k    sJ ||         j        d         t          t          j	        |d d |f                             k    sJ t          ||                             d          t          j        t          |                               t          t          j        ||                   ||                    d S )N   )r   r   axis)predict_probapredict_log_probanp
atleast_1dndimreshapeshapelenrangeuniquer   sumoneslog)clfXyproba	log_proba	n_outputs	n_samplesks           8lib/python3.11/site-packages/sklearn/tests/test_dummy.py_check_predict_probar'      sP   a  E%%a((I
aAv{{Jq'""
IAIA~~K	9 B BQx~a I----Qx~a C	!AAAqD'(:(:$;$;;;;;!%(,,A,"6"6AHHH!"&q"2"2IaLAAAAB B    c                    t          j        dgdgdgdgg          }t          j        g d          }t          |           }|                    ||           |                    |          }|j        |j        k    sJ t          j        ddgddgddgddgg          }t          |           }|                    ||           |                    |          }|j        |j        k    sJ d S )Nr   r      r   r   r   r+      r   arrayr   fitpredictr   r   r   r    esty_preds        r&   _check_behavior_2dr4   *   s    
1#sQC!%&&A
A
**CGGAqMMM[[^^F7fl"""" 	1a&1a&1a&1a&122A
**CGGAqMMM[[^^F7fl""""""r(   c                    t          j        dgdgdgdgg          }t          j        g dg dg dg dg          }t          |           }|                    ||           |                    |          }|j        |j        k    sJ d S )Nr   )r   r         r,   )r+   r   r   r+   r6   )r   r   r7   r6   r+   )r   r,   r,   r+   r   r-   r1   s        r&   _check_behavior_2d_for_constantr8   ;   s    
1#sQC!%&&A
///???OOO___UVVA
**CGGAqMMM[[^^F7fl""""""r(   c                     t          t          j        | |j        d         df          |           t          t          j        | |j        d         df          |           d S )Nr   r   )r   r   tiler   )	statisticy_learny_pred_learny_testy_pred_tests        r&   _check_equality_regressorr@   E   sW    bgi'-2BA1FGGVVVbgi&,q/11EFFTTTTTr(   c            	      B   dgdgdgdgg} g d}dD ]}t          |d          }|                    | |           t          |                    |           t	          j        t          |                                t          || |           |dk    rCt          |	                    | d         g          |j
                            d                     t          |	                    | d         g          |j
                            d          dk               d S )Nr   r*   most_frequentpriorstrategyrandom_staterD   r   r         ?)r   r/   r	   r0   r   r   r   r'   r   r   class_prior_r   )r   r    rF   r   s       r&   %test_most_frequent_and_prior_strategyrK   J   s+   
qcA3AA.  xa@@@13;;q>>273q66??;;;S!Q'''w%!!1Q4&))3+;+C+CG+L+L    &!!1Q4&))3+;+C+CG+L+Ls+R    r(   c                  H   dgdgdgdgg} g d}dgdgdgdgg}dD ]}t          |d          }t          |d          }|                    | |           |                    | |           t          |                    |           |                    |                      d S )Nr   r*   r   r+   rB   rE   r   r/   r	   r0   )r   y_1dy_2drF   clf_1dclf_2ds         r&   6test_most_frequent_and_prior_strategy_with_2d_column_yrR   ^   s     qcA3A<<DC!qcA3D. A A (CCC (CCC

1d

1d6>>!,,fnnQ.?.?@@@@A Ar(   c                     dgdgdgdgg} t          j        ddgddgddgddgg          }t          |           }dD ]}t          |d          }|                    | |           t          |                    |           t          j        t          j        |df          t          j	        |df          g                     t          || |           t          |           d S )Nr   r   r+   r,   )rD   rC   rE   )r   r.   r   r   r/   r	   r0   hstackr   zerosr'   r4   )r   r    r$   rF   r   s        r&   1test_most_frequent_and_prior_strategy_multioutputrV   n   s    
qcA3A
1a&1a&1a&1a&122AAI.    xa@@@1KKNNIrw	1~..)Q0H0HIJJ	
 	
 	
 	S!Q'''3   r(   c                    dggdz  }g d}t          d|           }|                    ||           dggdz  }|                    |          }t          j        |          t          t          |                    z  }t          |d         dd	           t          |d
         dd	           t          |||           d S )Nr   r6   )r   r+   r   r   r+   
stratifiedrE     r   333333?decimalr+   皙?	r   r/   r0   r   bincountfloatr   r   r'   global_random_seedr   r    r   r3   ps         r&   test_stratified_strategyrd      s    
	AA
<>P
Q
Q
QCGGAqMMM
A[[^^F
FeCFFmm+A!gq1111!gq1111a#####r(   c                 0   dggdz  }t          j        ddgddgddgddgddgg          }t          d|           }|                    ||           dggdz  }|                    |          }t          |j        d                   D ]~}t          j        |d d |f                   t          t          |                    z  }t          |d         dd	           t          |d         d
d	           t          |||           t          |           d S )Nr   r6   r+   r   rX   rE   rY   rZ   r[   r]   r   r.   r   r/   r0   r   r   r_   r`   r   r   r'   r4   rb   r   r    r   r3   r%   rc   s          r&   $test_stratified_strategy_multioutputrh      s!   
	A
1a&1a&1a&1a&1a&9::A
<>P
Q
Q
QCGGAqMMM
A[[^^F171: ( (Kqqq!t%%c!ff5AaD'15555AaD'15555S!Q''''sr(   c                    dggdz  }g d}t          d|           }|                    ||           dggdz  }|                    |          }t          j        |          t          t          |                    z  }t          |d         dd	           t          |d
         dd	           t          |||           d S )Nr   r7   r*   uniformrE   rY   r   rI   r[   r+   r^   ra   s         r&   test_uniform_strategyrk      s    
	AA
9;M
N
N
NCGGAqMMM
A[[^^F
FeCFFmm+A!c1----!c1----a#####r(   c                 *   dggdz  }t          j        ddgddgddgddgg          }t          d|           }|                    ||           dggdz  }|                    |          }t          |j        d                   D ]~}t          j        |d d |f                   t          t          |                    z  }t          |d         dd	           t          |d         dd	           t          |||           t          |           d S )
Nr   r7   r+   r   rj   rE   rY   rI   r[   rf   rg   s          r&   !test_uniform_strategy_multioutputrm      s   
	A
1a&1a&1a&1a&122A
9;M
N
N
NCGGAqMMM
A[[^^F171: ( (Kqqq!t%%c!ff5AaD#q1111AaD#q1111S!Q''''sr(   c                      dggdz  } g d}t          d          }|                    | |           t          |                    |           dgdz             d S )Nr   r6   )parisro   tokyo	amsterdamberlinrC   rF   ro   rM   r   r    r   s      r&   test_string_labelsru      sa    
	A:::A
?
3
3
3CGGAqMMMs{{1~~y1}55555r(   zy,y_test)r+   r   r   r   )r+   r+   r   r   r+   r   c                     t          d          }|                    d |            |                    d |          dk    sJ d S )NrC   rs   rI   )r   r/   score)r    r>   r   s      r&   test_classifier_score_with_Nonerx      sM     ?
3
3
3CGGD!99T6""c))))))r(   rF   )rX   rC   rD   rj   constantc                 >   g d}dggdz  }t          | |d          }|                    ||           |                    |          }dggdz  }t          | |d          }|                    ||           |                    |          }t          ||           d S )Nr   r+   r   r   r   r7   rF   rG   ry   r   )r   r/   r0   r	   )	rF   rb   r    X1clf1predictions1X2clf2predictions2s	            r&   +test_classifier_prediction_independent_of_Xr      s     	A#B(:Q  D 	HHROOO<<##L#B(:Q  D 	HHROOO<<##L|\22222r(   c                 R   t           j                            |           }dggdz  }|                    d          }t	                      }|                    ||           t          |                    |          t          j        |          gt          |          z             d S )Nseedr   r7   )
r   randomRandomStaterandnr   r/   r	   r0   meanr   rb   rG   r   r    regs        r&   test_mean_strategy_regressorr      s    9((.@(AAL
	A1A


CGGAqMMMs{{1~~

|c!ff'<=====r(   c                 (   t           j                            |           }|                    dd          }|                    dd          }t          j        |d                              d          }|                    dd          }|                    dd          }t                      }|                    ||           |                    |          }|                    |          }	t          |||||	           t          |           d S )Nr   
   r6   r   r   rH      )r   r   r   r   r   r   r   r/   r0   r@   r4   )
rb   rG   X_learnr<   r   X_testr>   r2   r=   r?   s
             r&   (test_mean_strategy_multioutput_regressorr      s    9((.@(AAL  R((G  Q''G77###++G44DB''FA&&F 

CGGGW;;w''L++f%%KdG\6;OOOsr(   c                      t                      } t          j        t                    5  |                     g            d d d            d S # 1 swxY w Y   d S )N)r   pytestraisesr   r0   )r   s    r&   test_regressor_exceptionsr     s    


C	~	&	&  B                 s   AAAc                 V   t           j                            |           }dggdz  }|                    d          }t	          d          }|                    ||           t          |                    |          t          j        |          gt          |          z             d S )Nr   r   r6   medianrs   )
r   r   r   r   r   r/   r	   r0   r   r   r   s        r&   test_median_strategy_regressorr     s    9((.@(AAL
	A1A
(
+
+
+CGGAqMMMs{{1~~	!~A'>?????r(   c                 ,   t           j                            |           }|                    dd          }|                    dd          }t          j        |d                              d          }|                    dd          }|                    dd          }t          d	          }|                    ||           |                    |          }|                    |          }	t          |||||	           t          |           d S )
Nr   r   r6   r   r   rH   r   r   rs   )r   r   r   r   r   r   r   r/   r0   r@   r4   )
rb   rG   r   r<   r   r   r>   r2   r=   r?   s
             r&   *test_median_strategy_multioutput_regressorr     s   9((.@(AAL  R((G  Q''GYwQ'''//88FB''FA&&F (
+
+
+CGGGW;;w''L++f%%Kfg|V[QQQsr(   c                    t           j                            |           }dggdz  }|                    d          }t	          dd          }|                    ||           t          |                    |          t          j        |          gt          |          z             t	          dd          }|                    ||           t          |                    |          t          j
        |          gt          |          z             t	          dd          }|                    ||           t          |                    |          t          j        |          gt          |          z             t	          dd          }|                    ||           t          |                    |          t          j        |d	
          gt          |          z             d S )Nr   r   r6   quantilerI   rF   r   r   g333333?   )q)r   r   r   r   r   r/   r	   r0   r   r   minmax
percentiler   s        r&    test_quantile_strategy_regressorr   4  s   9((.@(AAL
	A1A
*s
;
;
;CGGAqMMMs{{1~~	!~A'>???
*q
9
9
9CGGAqMMMs{{1~~q		{SVV';<<<
*q
9
9
9CGGAqMMMs{{1~~q		{SVV';<<<
*s
;
;
;CGGAqMMMs{{1~~a2(>(>(>'?#a&&'HIIIIIr(   c                 h   t           j                            |           }|                    dd          }|                    dd          }t          j        |d                              d          }t          j        |dd                              d          }|                    d	d          }|                    d	d          }t          d
d          }|                    ||           |	                    |          }	|	                    |          }
t          |||	||
           t          |           t          d
d          }|                    ||           |	                    |          }	|	                    |          }
t          |||	||
           t          |           d S )Nr   r   r6   r   r   rH   P   )r   r   r   r   rI   r   g?)r   r   r   r   r   r   r   r   r/   r0   r@   r4   )rb   rG   r   r<   r   quantile_valuesr   r>   r2   r=   r?   s              r&   ,test_quantile_strategy_multioutput_regressorr   K  s   9((.@(AAL  R((G  Q''GYwQ'''//88FmG!r:::BB7KKOB''FA&&F *s
;
;
;CGGGW;;w''L++f%%Kfg|V[QQQs *s
;
;
;CGGGW;;w''L++f%%K,   sr(   c                      dggdz  } dgdz  }t          dd           }d}t          j        t          |          5  |                    | |           d d d            d S # 1 swxY w Y   d S )Nr   r6   r   r   zIWhen using `strategy='quantile', you have to specify the desired quantilematchr   r   r   
ValueErrorr/   r   r    r2   err_msgs       r&   test_quantile_invalidr   l  s    
	A	
aA
*t
<
<
<CS  
z	1	1	1  1                 s   A  A$'A$c                      t          dd          } t          j        t                    5  |                     g g            d d d            d S # 1 swxY w Y   d S )Nr   r]   r   r   )r2   s    r&   "test_quantile_strategy_empty_trainr   x  s    
*s
;
;
;C	z	"	"  B                 s   AAAc                 .   t           j                            |           }dggdz  }|                    d          }t	          ddg          }|                    ||           t          |                    |          dgt          |          z             t	          dd          }|                    ||           t          |                    |          dgt          |          z             t          |j
        t           j                  rJ d S )Nr   r   r6   ry   +   rF   ry   )r   r   r   r   r   r/   r	   r0   r   
isinstancery   ndarrayr   s        r&    test_constant_strategy_regressorr   ~  s    9((.@(AAL
	A1A
*t
<
<
<CGGAqMMMs{{1~~tc!ff}555
*r
:
:
:CGGAqMMMs{{1~~tc!ff}555 #,
3333333r(   c                    t           j                            |           }|                    dd          }|                    dd          }|                    d          }|                    dd          }|                    dd          }t	          d|          }|                    ||           |                    |          }|                    |          }	t          |||||	           t          |           d S )Nr   r   r6   r   ry   r   )	r   r   r   r   r   r/   r0   r@   r8   )
rb   rG   r   r<   	constantsr   r>   r2   r=   r?   s
             r&   ,test_constant_strategy_multioutput_regressorr     s    9((.@(AAL  R((G  Q''G ""1%%IB''FA&&F *y
A
A
ACGGGW;;w''L++f%%Ki,TTT#C(((((r(   c                      dggdz  } g d}t          d          }|                    | |           |j        t          j        |          k    sJ d S )Nr   r6   r   r+   r7         r   rs   )r   r/   	constant_r   r   )r   r    r2   s      r&   test_y_mean_attribute_regressorr     sZ    
	AA
&
)
)
)CGGAqMMM=BGAJJ&&&&&&r(   c                      dggdz  } g d}t          d          }d}t          j        t          |          5  |                    | |           d d d            d S # 1 swxY w Y   d S )Nr   r6   r   ry   rs   z)Constant target value has to be specifiedr   )r   r   r   	TypeErrorr/   r   s       r&   &test_constants_not_specified_regressorr     s    
	AA
*
-
-
-C9G	y	0	0	0  1                 s   AA!$A!c                 \   t           j                            |           }|                    dd          }|                    dd          }t	          dg d          }d}t          j        t          |          5  |                    ||           d d d            d S # 1 swxY w Y   d S )	Nr   r   r6   ry   )r   r+   r,   r7   r   z1Constant target value should have shape \(5, 1\).r   )	r   r   r   r   r   r   r   r   r/   )rb   rG   r   r    r2   r   s         r&   (test_constant_size_multioutput_regressorr     s    9((.@(AAL2r""A2q!!A
*|||
D
D
DCBG	z	1	1	1  1                 s   =B!!B%(B%c                     dgdgdgdgg} g d}t          ddd          }|                    | |           t          |                    |           t	          j        t          |                                t          || |           dgdgdgdgg} g d}t          ddd          }|                    | |           t          |                    |           t	          j        dgdz                       t          || |           d S )	Nr   r+   r   r+   r+   ry   r   r|   )twooner   r   r   r7   )	r   r/   r	   r0   r   r   r   r'   r.   rt   s      r&   test_constant_strategyr     s   
qcA3AA
:A
J
J
JCGGAqMMMs{{1~~rws1vv777a###
qcA3A$$$A
:A
N
N
NCGGAqMMMs{{1~~rx!'<'<===a#####r(   c            
         dgdgdgdgg} t          j        ddgddgddgddgg          }t          |           }t          ddddg          }|                    | |           t          |                    |           t          j        t          j        |df          t          j	        |df          g                     t          || |           d S )Nr   r+   r,   r   ry   r|   )r   r.   r   r   r/   r	   r0   rT   r   rU   r'   )r   r    r$   r   s       r&   "test_constant_strategy_multioutputr     s    
qcA3A
1a&1a&1a&1a&122AAI
:AA
O
O
OCGGAqMMMA	27Iq>#:#:BHiQR^<T<T"UVV   a#####r(   zy, params, err_msgr   rG   zConstant.*has to be specifiedry   zConstant.*should have shapezmy-constantz.constant=my-constant.*Possible values.*\[1, 2]unknownz2constant=\[2, 'unknown'].*Possible values.*\[1, 2])zno-constantztoo-many-constantznot-enough-outputzsingle-outputzmulti-output)idsc                     dgdgdgdgg}t          dddi|}t          j        t          |          5  |                    ||            d d d            d S # 1 swxY w Y   d S )Nr   rF   ry   r    )r   r   r   r   r/   )r    paramsr   r   r   s        r&   !test_constant_strategy_exceptionsr     s    < qcA3A

8
8:
8
8
8C	z	1	1	1  1                 s   AAAc                      dgdgdgg} g d}g d}t          d                              | ||          }t          |j        ddg           d S )	Nr   r   )r   r   r   )皙?      ?r   rX   rs   gVUUUUU?g?)r   r/   r   rJ   )r   r    sample_weightr   s       r&   !test_classification_sample_weightr   	  se    
qcA3A		A#OOM
<
0
0
0
4
4Q=
I
ICc.I0FGGGGGr(   c            
         dggdz  } t          j        t          j        ddgddgddgddgddgg                    }t	          |           }t          ddddg          }|                    | |           |                    |           }t          j        |          sJ t          |
                                t          j        t          j        |df          t          j        |df          g                     d S )Nr   r6   r   r7   ry   r|   )sp
csc_matrixr   r.   r   r   r/   r0   issparser	   toarrayrT   r   rU   )r   r    r$   r   r3   s        r&   $test_constant_strategy_sparse_targetr     s    
	A
bhAAAAAGHHIIAAI
:AA
O
O
OCGGAqMMM[[^^F;v")RWi^%<%<bh	ST~>V>V$WXX    r(   c           
         dggdz  }t          j        t          j        ddgddgddgddgddgg                    }t	          d|           }t          j        t          d	          5  |                    ||           d d d            n# 1 swxY w Y   dggd
z  }|	                    |          }t          |j        d                   D ]}t          j        |d d |f                   t          t          |                    z  }t          |d         dd           t          |d         dd           t          |d         dd           d S )Nr   r6   r+   r   r7   rj   rE   z*the uniform strategy would not save memoryr   rY   gUUUUUU?r[   )r   r   r   r.   r   r   warnsUserWarningr/   r0   r   r   r_   r`   r   r   rg   s          r&   +test_uniform_strategy_sparse_target_warningr   !  s   
	A
bhAAAAAGHHIIA
9;M
N
N
NC	k)U	V	V	V  1               A[[^^F171: 4 4Kqqq!t%%c!ff5AaD%3333AaD%3333AaD%33333	4 4s   )BBBc           
         dggdz  }t          j        t          j        ddgddgddgddgddgg                    }t	          d|           }|                    ||           dggdz  }|                    |          }t          j        |          sJ |                                }t          |j
        d                   D ]}t          j        |d d |f                   t          t          |                    z  }t          |d         dd	           t          |d         d
d	           t          |d         d
d	           d S )Nr   r6   r7   r   rX   rE   rY   rZ   r[   g?)r   r   r   r.   r   r/   r0   r   r   r   r   r_   r`   r   r   rg   s          r&   &test_stratified_strategy_sparse_targetr   3  sM   
	A
bhAAAAAGHHIIA
<>P
Q
Q
QCGGAqMMM
A[[^^F;v^^F171: 6 6Kqqq!t%%c!ff5AaD'15555AaD'15555AaD'155555	6 6r(   c            
      
   dggdz  } t          j        t          j        ddgddgddgddgddgg                    }t	          |           }t          j        t          j        |df          t          j        |df          g          }dD ]v}t          |d          }|	                    | |           |
                    |           }t          j        |          sJ t          |                                |           wd S )Nr   r6   r   r,   r7   rB   rE   )r   r   r   r.   r   rT   r   rU   r   r/   r0   r   r	   r   )r   r    r$   
y_expectedrF   r   r3   s          r&   3test_most_frequent_and_prior_strategy_sparse_targetr   F  s    
	A
bhAAAAAGHHIIAAIBGYN33RXy!n5M5MNOOJ. 9 9xa@@@1Q{6"""""6>>++Z88889 9r(   r   c                 :   t           j                            |           }dgg|z  }|                    |          }|                    |          }t	          d                              |||          }|j        t          j        ||          k    sJ t	          d                              |||          }|j        t          ||d          k    sJ t	          dd	
                              |||          }|j        t          ||d          k    sJ d S )Nr   r   r   rs   )weightsr   g      I@r   gffffff?r   g     W@)	r   r   r   randr   r/   r   averager   )rb   r$   rG   r   r    r   r2   s          r&   "test_dummy_regressor_sample_weightr   U  s   9((.@(AAL
	A)$$A %%i00M
&
)
)
)
-
-aM
B
BC=BJq-@@@@@@@
(
+
+
+
/
/1m
D
DC=0M4HHHHHH
*t
<
<
<
@
@A}
U
UC=0M4HHHHHHHHr(   c                  *   t          j        dggdggdggg          } t          j        g d          }t          j        g d          }t                      }|                    | |           |                    |           }t          ||           d S )Nfoobarbazr+   r+   r+   )r   r.   r   r/   r0   r	   )r   r    r   clsr3   s        r&    test_dummy_regressor_on_3D_arrayr   f  s    
E7)ugY%	233A
A)))$$J


CGGAqMMM[[^^Fvz*****r(   c                  @   t          j        dggdggdggg          } g d}g d}dgdgdgg}t          d          }|                    | |           |                    |           }|                    |           }t          ||           t          ||           d S )Nr   r   r   r   r   rX   rs   )r   r.   r   r/   r0   r   r	   )r   r    r   y_proba_expectedr   r3   y_pred_probas          r&   !test_dummy_classifier_on_3D_arrayr   p  s    
E7)ugY%	233A		AJaS1#
<
0
0
0CGGAqMMM[[^^F$$Q''Lvz***|%566666r(   c                  8   dggdz  } t          j        g d          }t          j        g d          }t                      }|                    | |           |                    | d          }t          |          dk    sJ t          |d         |           d S )	Nr   r,   r   )r   r   r   T)
return_stdr+   r   )r   r.   r   r/   r0   r   r	   )r   r    y_std_expectedr   y_pred_lists        r&   test_dummy_regressor_return_stdr   }  s    
	A
AXiii((N


CGGAqMMM++aD+11K{q    {1~~66666r(   )r   r   r   r+   g      ?r7   c                     t                      }|                    d |            |                    d |          dk    sJ d S )Nr   )r   r/   rw   )r    r>   r   s      r&   test_regressor_score_with_Noner     sH     

CGGD!99T6""c))))))r(   )r   r   r   ry   c                 >   g d}dggdz  }t          | dd          }|                    ||           |                    |          }dggdz  }t          | dd          }|                    ||           |                    |          }t          ||           d S )Nr{   r   r7   gffffff?)rF   ry   r   r   )r   r/   r0   r	   )rF   r    r}   reg1r   r   reg2r   s           r&   *test_regressor_prediction_independent_of_Xr    s    A#B8a#FFFDHHROOO<<##L#B8a#FFFDHHROOO<<##L|\22222r(   c                     g d}t          j        d          }t          | dd          }|                    ||                              |          }|j        t           j        k    sJ d S )Nr{   r7   r   r|   )r   rU   r   r/   r   dtypefloat64)rF   r    r   modelprobass        r&   test_dtype_of_classifier_probasr	    sh     	A
AXAJJJEYYq!__**1--F<2:%%%%%%r(   )r   )Cnumpyr   r   scipy.sparsesparser   sklearn.baser   sklearn.dummyr   r   sklearn.exceptionsr   sklearn.utils._testingr   r   r	   r
   sklearn.utils.statsr   r'   r4   r8   r@   rK   rR   rV   rd   rh   rk   rm   ru   markparametrizer.   rx   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   	transposer   r   r   r   r   r   r   r   r   r   r   r  r	  r   r(   r&   <module>r     s                    9 9 9 9 9 9 9 9 - - - - - -            5 4 4 4 4 4 B B B0# # #"# # #U U U
  (A A A      "$ $ $  &$ $ $  $6 6 6 	|||$BHq!fq!fq!fq!f566BHq!fq!fq!fq!f566	
	 	* *	 	* OOO 3 3 3&> > >  *  @ @ @  *J J J.  B	 	 	  4 4 4$) ) ),' ' '    $ $ $"$ $ $ 	*,KL	
QF+-JKBL,,,566O)	
 LL'=	
 BL,,,566!Y(B	
&	 	 	+   : ; :H H H  4 4 4$6 6 6&9 9 9I I I I"+ + +
7 
7 
7
7 
7 
7 	vz"	Aq6Aq6Aq6Aq62	3	3tTl^a5GH * * * %O%O%OPP3 3 QP3 OOO & & & & &r(   