
    riJ                       d dl Z d dlZd dl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 d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZm Z  d d	l!m"Z" d d
l#m$Z$ d dl%m&Z& d dl'm(Z(m)Z)m*Z*m+Z+m,Z, d dl-m.Z. d dl/m0Z0 d dl1m2Z2m3Z3m4Z4 eeeegZ5ddZ6d Z7d Z8d Z9d Z:d Z;ejx                  j{                  dddg      d        Z>d Z?d Z@d ZAd ZBd ZCd ZDd  ZEd! ZFd" ZGd# ZHd$ ZIejx                  j{                  d% ej                  g d&      g d'f ej                  g d&      dfg d(g d)fg d(dfg      d*        ZKejx                  j{                  d% ej                  g d+      g d'f ej                  g d,      g d-fg      d.        ZLejx                  j{                  d% ej                  g d/      dfg d0dfg d/g d'fg d0g d)fg      d1        ZMejx                  j{                  d2g d3      d4        ZNd5 ZOejx                  j{                  d6d7 ej                  g d/      g d8fd7 ej                  g d0      g d9fd: ej                  g d+      dfd; ej                  g d0      g d<fd= ej                  g d/      d d>gfd= ej                  g d0      d?d@gfdA ej                  g d/      g dBfdA ej                  g d0      g dCfdD ej                  g dE      g d)fdD ej                  g dC      g d)fdD ej                  g dB      g d'fg      ejx                  j{                  dFdGdHg      dI               ZPejx                  j{                  dJdKdLdGdMfdNdLdHdMfdOdGg dPfdQdGdRdSfdTi fg      dU        ZQdV ZRejx                  j{                  dWe5      dX        ZSejx                  j{                  dWe5      dY        ZTejx                  j                  dZ      ejx                  j{                  dWeeg      ejx                  j{                  d[d\d]g      d^                      ZVejx                  j{                  dWe5      d_        ZWejx                  j{                  dddg      d`        ZXda ZYejx                  j{                  dddg      db        ZZdc Z[dd Z\de Z]df Z^dg Z_dh Z`ejx                  j{                  dig djg dkd gd gfg djg dld gd gfg djg dmd gd gfg djg dnd gd gfg dog dkg dpg dqfg dog dlg dpg dqfg dog dmg dpg dqfg dog dng dpg dqfg dqg dkdrgd gfg dqg dld gd gfg dqg dmd gd gfg dqg dnd gd gfg dsg dkg dtg dufg dsg dlg dtg dufg dsg dmg dtg dufg dsg dng dtg dufg dvg dkg dwg dxfg dvg dlg dwg dxfg dvg dmg dwg dxfg dvg dng dwg dxfg      dy        Zaejx                  j{                  g dzg dsg dkg d{g dudfg dsg dkg d|g dqdfg dsg dlg d{g dudfg dsg dlg d|g dqdfg dvg dkg dwg dxdfg dvg dkg dwg dxdfg dvg dlg dwg dxdfg dvg dlg dwg dxdfg      d}        Zbejx                  j{                  did>d gdRdRgd>d gd d>gfd d>gdRdRgd>d gd d>gfg djg d~dRd gd d>gfg dog d~dRd gd d>gfg dqg d~d gd gfg dsg d~g dwg dqfg dvg d~g dwg dkfg dwg d~g dwg dkfg      d        Zcd Zdejx                  j{                  dg d      d        Zeejx                  j{                  dg dg dg dg dg dg      d        Zfejx                  j{                  dd d>gg dkdfg dqd dRgdfg dg dkdfg dg dkdfg dg ddfg      d        Zgd Zhd Zid Zjd Zkd Zld Zmd Zn	 ddZoejx                  j{                  deiemelejf      ejx                  j{                  deenf      d               Zpd Zqejx                  j{                  dd      ejx                  j{                  dd      ejx                  j{                  d erd>            d                      Zsd Ztd Zud Zvejx                  j{                  dg dvg d~fg dvg d~gfg dvgg d~fg      d        Zwd Zxejx                  j{                  de0      d        Zyd Zzd Z{d Z|d Z}d Z~d Zd Zd Zejx                  j{                  dddg      d        Zd Zd Zd Zd Zejx                  j{                  dg dBd>dfg dBddRfg dBddfg      d        Zejx                  j{                  d ej                  g dĢ      d>d>f ej                  g dŢ      d>dRf ej                  g dŢ      dd>f ej                  g dƢ      d>d>f ej                  g dǢ      d>dRf ej                  g dǢ      dd>fg      dȄ        Zejx                  j{                  d ej                  g dʢ      dg dBf ej                  g du      dRg dBf ej                  g dˢ      dRg dBf ej                  g d̢      dg d͢fg      ejx                  j{                  dddg      dτ               ZdЄ Zejx                  j{                  dg dBd>dfg dBddRfg dBdd>fg      dф        Zejx                  j{                  dg dBdfg dBdfg      dԄ        Zejx                  j{                  dg d֢g dעg dآg d٢g dڢgddfg dBg dעg dآg d٢g dڢgddfg dݢg dעg dآg d٢g dڢgg d0dfg dݢg dעg dآg d٢g dڢgg d<dfg dg dעg dآg d٢g dڢgg dBdfg dg dעg dآg d٢g dڢgg ddfd d>gg dg dgddfg      d        Zejx                  j{                  de0      d        Zejx                  j{                  deeeeg      ejx                  j{                  dg d      d               Zd Zy)    N)stats)datasets)make_multilabel_classification)UndefinedMetricWarning)LogisticRegression)accuracy_scoreaucaverage_precision_scoreconfusion_matrix_at_thresholdscoverage_error	dcg_score	det_curve%label_ranking_average_precision_scorelabel_ranking_loss
ndcg_scoreprecision_recall_curveroc_auc_score	roc_curvetop_k_accuracy_score)_dcg_sample_scores_ndcg_sample_scores)train_test_split)label_binarize)_sparse_random_matrix)_convert_containerassert_allcloseassert_almost_equalassert_array_almost_equalassert_array_equal)softmax)CSR_CONTAINERS)check_arraycheck_consistent_lengthcheck_random_stateFc                 ~   | t        j                         } | j                  }| j                  }|r||dk     ||dk     }}|j                  \  }}t        j                  |      }t        d      }|j                  |       ||   ||   }}t        |dz        }t
        j                  j                  d      }t
        j                  ||j                  |d|z        f   }t        d      }	|	j                  |d| |d|       j!                  ||d       }
|r	|
dddf   }
|	j#                  ||d       }||d }|||
fS )zMake some classification predictions on a toy dataset using an SVC

    If binary is True restrict to a binary classification problem instead of a
    multiclass classification problem
    N   %   r      random_state   )r   	load_irisdatatargetshapenparanger$   shuffleintrandomRandomStatec_randnr   fitpredict_probapredict)datasetbinaryXy	n_samples
n_featuresprnghalfclfy_scorey_predy_trues                `/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/sklearn/metrics/tests/test_ranking.pymake_predictionrI   :   sE    $$&AAQx1q51GGIz
		)A
R
 CKKNQ41qAy1}D ))


"C
a9cJ&6778A !
,Cggah%4)77$%AG !Q$-[[45"FtuXF67""    c                    t        j                  |       d   }|| |k(     }|| |k7     }|j                  dd      |j                  dd      z
  }t        j                  |dkD        }|t	        t        |      t        |      z        z  S )zKAlternative implementation to check for correctness of
    `roc_auc_score`.r+   r   )r0   uniquereshapesumfloatlen)rG   rE   	pos_labelposnegdiff_matrix	n_corrects          rH   _aucrW   j   s     		&!!$I &I%
&C
&I%
&C++a$s{{2q'99K{Q'IuSXC0111rJ   c                 `   t        j                  |       d   }t        j                  | |k(        }t        j                  |      ddd   }||   }| |   } d}t	        t        |            D ]<  }| |   |k(  sd}t	        d|dz         D ]  }| |   |k(  s|dz  } ||dz   z  }||z  }> ||z  S )a>  Alternative implementation to check for correctness of
    `average_precision_score`.

    Note that this implementation fails on some edge cases.
    For example, for constant predictions e.g. [0.5, 0.5, 0.5],
    y_true = [1, 0, 0] returns an average precision of 0.33...
    but y_true = [0, 0, 1] returns 1.0.
    r+   NrL   r         ?)r0   rM   rO   argsortrangerQ   )	rG   rE   rR   n_posorderscoreiprecjs	            rH   _average_precisionrb   y   s     		&!!$IFF6Y&'EJJw"%EenGE]FE3w<  	!9	! D1a!e_  !9	)CKD  AGODTME	 5=rJ   c                     t        | |      \  }}}t        t        |            }t        t        |            }d}t        dt	        |            D ]  }|||   ||   ||dz
     z
  z  z  } |S )ao  A second alternative implementation of average precision that closely
    follows the Wikipedia article's definition (see References). This should
    give identical results as `average_precision_score` for all inputs.

    References
    ----------
    .. [1] `Wikipedia entry for the Average precision
       <https://en.wikipedia.org/wiki/Average_precision>`_
    r   r+   )r   listreversedr[   rQ   )rG   rE   	precisionrecall	thresholdaverage_precisionr_   s          rH   _average_precision_slowrj      s     $:&'#J IvyXi()I(6"#F1c)n% HYq\VAYA-FGGHrJ   c                     d } || ||      \  }}t        ||      }d}|}d||z
  z  ||z   z  }	||z
  }
dd||	z
  |
|	z
  z  z   z  S )zcAlternative implementation to check for correctness of `roc_auc_score`
    with `max_fpr` set.
    c                 .   t        | |      \  }}}|||k     }t        j                  ||      }|||k     }t        j                  ||kD        }|dz
  }	||	   ||   g}
||	   ||   g}t        j                  |t        j                  ||
|            }||fS )Nr+   )r   r0   appendargmaxinterp)rG   	y_predictmax_fprfprtpr_new_fprnew_tpridx_outidx_inx_interpy_interps               rH   _partial_rocz,_partial_roc_auc_score.<locals>._partial_roc   s    	2S!cWn%))GW-cWn%))C'M*1KW.KW.))GRYYw(%KL!!rJ   r         ?r+   )r	   )rG   rp   rq   r{   ru   rv   partial_aucfpr1fpr2min_areamax_areas              rH   _partial_roc_auc_scorer      ss    

" $FIw?GWgw'K DDdTk"dTk2Hd{H!{X-(X2EFFGGrJ   c                    t         j                  j                  |       }d}|j                  ddd      }|j	                  d      }t        j
                  |      }||z
  }t        ||      \  }}}	}
}t        |      t        |      cxk(  r-t        |	      cxk(  rt        |
      cxk(  rt        |      k(  sJ  J t        |
|	z   |       t        ||z   |       t        ||z   |	z   |
z   |       y)z.Smoke test for confusion_matrix_at_thresholds.d   r   r&   sizeN)	r0   r4   r5   randintuniformrO   r   rQ   r   )global_random_seedrB   r?   rG   rE   r\   n_negtnsfpsfnstps
thresholdss               rH   #test_confusion_matrix_at_thresholdsr      s    
))

 2
3CI[[AC[(Fkksk#GFF6NEE%CFG%T"Cc3
s8s3xJ3s8Js3xJ3z?JJJJJC#Iu%C#Iu%C#IOc)95rJ   dropTc                 ,   t        d      \  }}}t        ||      }t        |||       \  }}}t        ||      }t	        ||d       t        |t        ||             |j                  |j                  k(  sJ |j                  |j                  k(  sJ y )NTr<   drop_intermediater&   decimal)rI   rW   r   r	   r   r   r   r/   )	r   rG   rt   rE   expected_aucrr   rs   r   roc_aucs	            rH   test_roc_curver      s     )5FAw(L$VWMCj#smGg|Q?vw!?@99		!!!99
(((((rJ   c                  `   t         j                  j                  d      } t        j                  dgdz  dgdz  z         }| j	                  dd      }t        ||d      \  }}}|d   dk(  sJ |d	   dk(  sJ |j                  |j                  k(  sJ |j                  |j                  k(  sJ y )
Nr   2   r+      r   r   Tr   rL   )r0   r4   r5   arrayr   r   r/   )rB   rG   rF   rr   rs   thrs         rH   test_roc_curve_end_pointsr      s     ))


"CXXqcBh!r)*F[[[%FffEMCcq6Q;;r7a<<99		!!!99		!!!rJ   c                  l   t        d      \  } }}t        | |      \  }}}g }|D ]I  }t        j                  ||k\  | z        }t        j                  |       }	|j	                  d|z  |	z         K t        ||d       |j                  |j                  k(  sJ |j                  |j                  k(  sJ y )NTr   rY   r&   r   )rI   r   r0   rO   rm   r   r/   )
rG   rt   rE   rr   rs   r   tpr_correctttprA   s
             rH   test_roc_returns_consistencyr      s     )5FAw$VW5Cj K )VVW\V+,FF6N38a<() c;:99		!!!99
(((((rJ   c                      t        d      \  } }}t        j                  t              5  t	        | |       d d d        y # 1 sw Y   y xY w)NFr   )rI   pytestraises
ValueErrorr   )rG   rt   rE   s      rH   test_roc_curve_multir     s>    (6FAw	z	" #&'"# # #s   A  A	c                      t        d      \  } }}t        | |dz
        \  }}}t        ||      }t        |dd       |j                  |j                  k(  sJ |j                  |j                  k(  sJ y )NTr   r|   ?r&   r   )rI   r   r	   r   r/   )rG   rt   rE   rr   rs   r   r   s          rH   test_roc_curve_confidencer     so    (5FAw$VWs];Cj#smGgtQ799		!!!99
(((((rJ   c                     t        d      \  } }}t        j                  | j                        }t	        | |      \  }}}t        ||      }t        |dd       |j                  |j                  k(  sJ |j                  |j                  k(  sJ t        j                  | j                        }t	        | |      \  }}}t        ||      }t        |dd       |j                  |j                  k(  sJ |j                  |j                  k(  sJ t	        | |      \  }}}t        ||      }t        |dd       |j                  |j                  k(  sJ |j                  |j                  k(  sJ y )NTr   r|   r&   r   g(\?)rI   r0   onesr/   r   r	   r   zeros)rG   predrE   trivial_predrr   rs   r   r   s           rH   test_roc_curve_hardr     sE   +48FD' 776<<(L$V\:Cj#smGgtQ799		!!!99
((((( 88FLL)L$V\:Cj#smGgtQ799		!!!99
((((( %VT2Cj#smGgtQ799		!!!99
(((((rJ   c                     g d} g d}d}t        j                  t        |      5  t        | |      \  }}}d d d        t	        t        j                  t              t
        j                               |j                  j                  k(  sJ |j                  |j                  k(  sJ d}t        j                  t        |      5  t        | D cg c]  }d|z
  	 c}|      \  }}}d d d        t	        |t        j                  t        |      t
        j                               |j                  |j                  k(  sJ |j                  |j                  k(  sJ y # 1 sw Y   ,xY wc c}w # 1 sw Y   xY w)N)
r+   r+   r+   r+   r+   r+   r+   r+   r+   r+   )
r   r+   r   r+   r   r+   r   r+   r   r+   INo negative samples in y_true, false positive value should be meaninglessmatchHNo positive samples in y_true, true positive value should be meaninglessr+   )
r   warnsr   r   r   r0   fullrQ   nanr/   )rG   rF   expected_messagerr   rs   r   xs          rH   test_roc_curve_one_labelr   6  sH   +F+F 	T  
,4D	E 9(8S*9 sBGGC
ORVV<=99		!!!99
((((( 	S  
,4D	E J()@A!a%)@&IS*J sBGGC
ORVV<=99		!!!99
(((((#9 9 *AJ Js)   E!	
E3E.
E3!E+.E33E<c                     ddg} ddg}t        | |      \  }}}t        | |      }t        |g d       t        |g d       t        |d       ddg} ddg}t        | |      \  }}}t        | |      }t        |g d       t        |g d       t        |d       ddg} ddg}t        | |      \  }}}t        | |      }t        |ddg       t        |ddg       t        |d       ddg} ddg}t        | |      \  }}}t        | |      }t        |g d       t        |g d       t        |d       ddg} ddg}t        | |      \  }}}t        | |      }t        |ddg       t        |ddg       t        |d       ddg} dd	g}d
}t	        j
                  t        |      5  t        | |      \  }}}d d d        t        |g d       t        |t        j                  t        j                  t        j                  g       d}t	        j
                  t        |      5  t        | |      }d d d        t        j                        sJ ddg} dd	g}d}t	        j
                  t        |      5  t        | |      \  }}}d d d        t        |t        j                  t        j                  t        j                  g       t        |g d       d}t	        j
                  t        |      5  t        | |      }d d d        t        j                  |      sJ t        j                  ddgddgg      } t        j                  ddgddgg      }t	        j
                  t        |      5  t        | |d       d d d        t	        j
                  t        |      5  t        | |d       d d d        t        t        | |d      d       t        t        | |d      d       t        j                  ddgddgg      } t        j                  ddgddgg      }t	        j
                  t        |      5  t        | |d       d d d        t	        j
                  t        |      5  t        | |d       d d d        t        t        | |d      d       t        t        | |d      d       t        j                  ddgddgg      } t        j                  ddgddgg      }t        t        | |d      d       t        t        | |d      d       t        t        | |d      d       t        t        | |d      d       t        j                  ddgddgg      } t        j                  ddgddgg      }t        t        | |d      d       t        t        | |d      d       t        t        | |d      d       t        t        | |d      d       y # 1 sw Y   )xY w# 1 sw Y   xY w# 1 sw Y   uxY w# 1 sw Y   	xY w# 1 sw Y   xY w# 1 sw Y   rxY w# 1 sw Y   xY w# 1 sw Y   xY w)Nr   r+   r   r   r+   r   r+   r+   rY           r|         ?      ?r   r   r   r|   rY   OOnly one class is present in y_true. ROC AUC score is not defined in that case.r   macroaverageweightedsamplesmicro)r   r   r   r   r   r   r   r0   r   mathisnanr   )rG   rE   rs   rr   rt   r   r   r	   s           rH   test_roc_curve_toydatar   Q  s   VF!fGFG,KCaFG,Gc9-c9-%VF!fGFG,KCaFG,Gc9-c9-%VF!fGFG,KCaFG,GcAq6*cAq6*%VF!fGFG,KCaFG,Gc9-c9-%VFCjGFG,KCaFG,GcAq6*cAq6*% VFTlG 	S  
,4D	E 10S!1c?3cBFFBFFBFF#;<	5  
,4D	E -FG,-::c?? VFTlG 	T  
,4D	E 10S!1cBFFBFFBFF#;<c?3	5  
,4D	E -FG,-::c?? XX1v1v&'FhhAA'(G	,4D	E 8fgw78	,4D	E ;fgz:;fgyI3OfgwGMXX1v1v&'FhhAA'(G	,4D	E 8fgw78	,4D	E ;fgz:;fgyI3OfgwGMXX1v1v&'FhhAA'(GfgwGKfgzJANfgyI1MfgwGKXX1v1v&'Fhhc
S#J/0GfgwGMfgzJCPfgyI3OfgwGM}1 1- -1 1- -8 8; ;8 8; ;s`   "WW'0W4-X*XX4X(&X5W$'W14W>XXX%(X25X?c            	          g d} g d}t        | |d      \  }}}t        |t        j                  dddg       g d} g d	}t        | |d      \  }}}t        |t        j                  dd
dddg       y )Nr   r   r   r   r+   r+   r   皙?r|   333333?ffffff?rY   Tr   rY   r   r   r   r   r   r   r   r   r   r+   r+   r+   r+   r+   r+   r   皙?r   r   r   皙?r   r   r   r   r   r   rY   r   r   )r   r   r0   inf)rG   rE   rs   rr   r   s        rH    test_roc_curve_drop_intermediater     st    F,G$VWMCjj2663S*AB 5FOG$VWMCjj2663S#s*KLrJ   c                     g d} g d}t        j                  dd      }t        | ||      \  }}}t        j                  |      dk  j	                         dk(  sJ t        j                  |      dk  j	                         dk(  sJ y )Nr   r   r+   r+   r+   )r   r   333333?皙?r|   r      sample_weightr   )r0   repeatr   diffrO   )rG   rE   r   rr   rs   rt   s         rH   !test_roc_curve_fpr_tpr_increasingr     sz     F'GIIc1%MFG=IKCaGGCL1!!#q(((GGCL1!!#q(((rJ   c                  0   ddg} ddg}t        t        | |      d       ddg} ddg}t        t        | |      d       g d} g d}t        t        | |      d       ddg} ddg}t        t        | |      d       g d} g d}t        t        | |      d       y )Nr   r+   r|   r+   r   r   r   r   r|   r+   )r   r	   )r   r>   s     rH   test_aucr     s    	
AA	
AAc!Qi-	
AA	
AAc!Qi-AAc!Qi-	
AA	
AAc!Qi+AAc!Qi-rJ   c                     t        j                  t              5  t        g dddg       d d d        t        j                  t              5  t        dgdg       d d d        g d} g d}dj	                  t        j                  |             }t        j                  t        t        j                  |            5  t        | |       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   y xY w)	Nr   r   r   r   )r&   r+   r      )r            z+x is neither increasing nor decreasing : {}r   )	r   r   r   r	   formatr0   r   reescape)r   r>   error_messages      rH   test_auc_errorsr     s    	z	" )Oc3Z() 
z	" SEC5 	AAAHHRSUM	z=)A	B Aq	 ) )  s#   CC=C+CC(+C4zy_true, labels)r   r+   r   r&   r   r+   r&   )abr   c)r   r   r   c           	      v   t        j                  g dg dg dg dg      }t        g dg d      }t        g dg d      }||z   d	z  }t        g d
g d      }t        g dg d      }||z   d	z  }t        ddgddg      }	t        ddgddg      }
|	|
z   d	z  }||z   |z   dz  }t        t        | ||d      |       |||g}g d}t        j                  ||      }t        t        | ||dd      |       d}t        j                  t        |      5  t        | ||dd        d d d        y # 1 sw Y   y xY w)Nr   r   r   r   r   r   ffffff?r|   333333?r   r   r   r+   r   r+   )r   r   r   r   r+   r   )r   r   r|   r&   r+   r+   r   )r   r   r   r   )r   r   r   r+   r   r   r   r   r   r   ovolabelsmulti_class)r   r   r|   )weightsr   r  r  r   z6average=None is not implemented for multi_class='ovo'.r   )r0   r   r   r   r   r   r   NotImplementedError)rG   r  y_scoresscore_01score_10average_score_01score_02score_20average_score_02score_12score_21average_score_12ovo_unweighted_scorepair_scores
prevalenceovo_weighted_scorer   s                    rH   #test_multiclass_ovo_roc_auc_toydatar    si    xx	/+<mLH Y(89HY8H 8+q0 Y7HY(89H 8+q0 aVc3Z0HaVc3Z0H 8+q0 -/??BRRVWWfhv5I $%57GHK#JKDHV
	
 		 MM	*-	@ Xfhv5RVWX X Xs   D//D8)r   r&   r   r&   )r   dr   r  )r   r   r  c           	          t        j                  g dg dg dg dg      }t        g dg d      }t        g dg d      }||z   d	z  }t        t        | ||d
      |       t        t        | ||d
d      |       y )N)r   r   r   )r   r   r   )皙?r   ?)r   r   r   )r+   r   r+   r   )r   r   r  r   )r   r+   r   r+   )r   r   r  r   r&   r  r  r   r  r0   r   r   r   )rG   r  r	  r
  r  	ovo_scores         rH   *test_multiclass_ovo_roc_auc_toydata_binaryr  <  s     xx	/+<oNH \+@AH\+@AHH$)Ifhv5I9
 HV
	
 		rJ   )r   r+   r&   r&   )r   r   r   r   c           	         t        j                  g dg dg dg dg      }t        g d|d d df         }t        g d|d d df         }t        g d	|d d d
f         }t        t        | |d|d       |||g       ||z   |z   dz  }t        t        | |d|      |       |dz  |dz  z   |dz  z   }t        t        | |d|d      |       y )NrY   r   r   )r   r|   r   )r   r   r   )r   r   r   r+   r   r   r   r   )r   r+   r   r   r+   r   r   r+   r+   r&   ovr)r  r  r   g      @)r  r  r   r|   r   r  )rG   r  r	  out_0out_1out_2result_unweightedresult_weighteds           rH   #test_multiclass_ovr_roc_auc_toydatar)  `  s     xx	/?OLH
 ,A7E,A7E,A7EfhE&RVW	u .#5fhE&I dlUT\1ECK?OH%
	
 		rJ   zmulti_class, average))r#  r   )r#  r   )r  r   c                 B   t        j                  g d      }g dg dg dg dg}t        t        ||| |      d       g dg dg dg dg}t        ||| |      dk  sJ dt        j                  d	      z  }t        ||| |      t        j                  d
      k(  sJ y )N)r   r+   r&   r   )r   r   r   rY   )r   rY   r   r   )r   r   rY   r   )r   皙?r+  r   r  r   rY   r   )r   r   r|   )r0   r   r   r   r   r   approx)r  r   rG   	y_perfecty_imperfecty_chances         rH   0test_perfect_imperfect_chance_multiclass_roc_aucr1    s     XXl#F 	 	I fi['R 		K 	fk{GT
		
 bggfo%Hk7	s	  rJ   c           
         | }t         j                  j                  g dd|      }t        j                  |D cg c]2  }t         j
                  j                  d||      j                         4 c}      }t        |g d      }t        |j                         |j                               \  }}}t        ||      }	t        ||dd	
      }
|	t        j                  |
      k(  sJ y c c}w )N)       @rY   r|     )r   r*   r+   )nrA   r*   r   )classesr#  r   r,  )r   	dirichletrvsr0   asarraymultinomialrn   r   r   ravelr	   r   r   r-  )r   seedrF   y_pred_irG   y_onehotrr   rs   rt   roc_auc_by_handroc_auc_autos              rH   test_micro_averaged_ovr_roc_aucrA    s    D
 __  t$ OFZZ #	
 !!A!ELLN	
F fi8HHNN,flln=KCa#smO UGTLfmmL9999	
s   7C)zmsg, y_true, labelsz!Parameter 'labels' must be uniquer   r&   r   )r   r   r   zKNumber of classes in y_true not equal to the number of columns in 'y_score'z"Parameter 'labels' must be ordered)r   r   r   zMNumber of given labels, 2, not equal to the number of columns in 'y_score', 3r+   r   r   zMNumber of given labels, 4, not equal to the number of columns in 'y_score', 3)r   r+   r&   r   )r   r   r   r  z2'y_true' contains labels not in parameter 'labels')r   r   r   er  r  r#  c                     t        j                  g dg dg dg dg      }t        j                  t        |       5  t        ||||       d d d        y # 1 sw Y   y xY w)Nr   r   r   r   r   r  )r0   r   r   r   r   r   )msgrG   r  r  r	  s        rH   *test_roc_auc_score_multiclass_labels_errorrF    sW    R xx	/+<mLH 
z	- Pfhv;OP P Ps   AAzmsg, kwargszLaverage must be one of \('macro', 'weighted', None\) for multiclass problemsr   )r   r  zUaverage must be one of \('micro', 'macro', 'weighted', None\) for multiclass problemszksample_weight is not supported for multiclass one-vs-one ROC AUC, 'sample_weight' must be None in this case)r  r   z|Partial AUC computation not available in multiclass setting, 'max_fpr' must be set to `None`, received `max_fpr=0.5` insteadr|   )r  rq   z'multi_class must be in \('ovo', 'ovr'\)c                     t        d      }|j                  dd      }t        |      }|j                  ddd      }t	        j
                  t        |       5  t        ||fi | d d d        y # 1 sw Y   y xY w)N     r   r   r   r   )r$   randr    r   r   r   r   r   )rE  kwargsrB   rE   y_probrG   s         rH   #test_roc_auc_score_multiclass_errorrM    sn    N S
!Chhr1oGWF[[AB['F	z	- 0ff//0 0 0s   A//A8c                  4   t        d      } | j                  d      }t        j                  dd      }d}t	        j
                  t        |      5  t        ||       d d d        t        j                  dd      }t	        j
                  t        |      5  t        ||       d d d        t        j                  ddd      }t	        j
                  t        |      5  t        ||       d d d        y # 1 sw Y   xY w# 1 sw Y   ^xY w# 1 sw Y   y xY w)NrH  
   r3   dtyper   r   rL   )
r$   rJ  r0   r   r   r   r   r   r   r   )rB   rF   rG   warn_messages       rH   test_auc_score_non_binary_classrS  I  s     S
!CXXb\FXXb&F	5  
,L	A &ff%&WWRu%F	,L	A &ff%&WWR5)F	,L	A &ff%& && && && &s$   C6D D6C?DD
curve_funcc                     t        d      }|j                  ddd      }|j                  d      }d}t        j                  t
        |      5   | ||       d d d        y # 1 sw Y   y xY w)NrH  r   r   rO  r   z"multiclass format is not supportedr   )r$   r   rJ  r   r   r   )rT  rB   rG   rF   rE  s        rH   4test_confusion_matrix_at_thresholds_multiclass_errorrV  ^  s_    
S
!C[[AB['FXXb\F
.C	z	- #66"# # #s   
A!!A*c                    d}t        j                  t        |      5   | t        j                  ddgd      ddg       d d d        t        j                  t        |      5   | t        j                  ddgt
              ddg       d d d        g d	} | g d
|      } | g d|      }t        ||      D ]%  \  }}t        j                  j                  ||       ' y # 1 sw Y   xY w# 1 sw Y   dxY w)Nzy_true takes value in {'a', 'b'} and pos_label is not specified: either make y_true take value in {0, 1} or {-1, 1} or pass pos_label explicitly.r   r   r   z<U1rP  r   rY   )r   rY   r   gzG?)r   r+   r+   r   )r   rY   rY   r   )	r   r   r   r0   r   objectziptestingr   )rT  rE  rF   	int_curvefloat_curveint_curve_partfloat_curve_parts          rH   6test_confusion_matrix_at_thresholds_implicit_pos_labelr_  h  s    
	  
z	- B288S#Je4sCjAB 
z	- C288S#Jf5SzBC
 #F<0I16:K,/	;,G E((


"">3CDEB BC Cs   #C"$'C."C+.C7z.ignore:Support for labels represented as byteslabels_typerd   r   c                     t        ddg|      }d}t        j                  t        |      5   | |ddg       d d d        y # 1 sw Y   y xY w)N   a   bz8Support for labels represented as bytes is not supportedr   r   rY   )r   r   r   	TypeError)rT  r`  r  rE  s       rH   <test_confusion_matrix_at_thresholds_implicit_bytes_pos_labelre    sJ      tk:F
DC	y	, '6C:&' ' 's   AA
c                     g d}g d}g d} | |||      } | |d d |d d |d d       }t        ||      D ]  \  }}t        ||        y )Nr   )r   r   r   r   r|   )r+   r+   r+   r|   r   r   rL   )rY  r   )rT  rG   rE   r   result_1result_2arr_1arr_2s           rH   6test_confusion_matrix_at_thresholds_zero_sample_weightrk    sf    F'G%M&'GH&"+ws|=QTRTCUVHHh/ &uu%&rJ   c                    t        d      \  }}}t        |||        t        |dd  |dd  |       \  }}}|d   dk(  sJ |d   |dd  j                         k(  sJ d|t	        j
                  |dk(        <   |j                         }t        |||        t        ||       g d}g d	}	t        ||	|       \  }}}| r+t        |g d
       t        |g d       t        |g d       n*t        |g d       t        |g d       t        |g d	       |j                  |j                  k(  sJ |j                  |j                  dz   k(  sJ y )NTr   r+   r   r   rY   rL   r+   r   r   r+   )r+   r&   r   r   )r|   QUU?rY   rY   )rY   r|   r|   r   )r+   r&   r   )r|   rn  r|   rY   rY   )rY   r|   r|   r|   r   )
rI   _test_precision_recall_curver   meanr0   wherecopyr   r   r   )
r   rG   rt   rE   rA   rr   y_true_copyr  predict_probass
             rH   test_precision_recall_curverv    s<   (5FAw $7 %VABZPTUGAq!Q43;;Q46!":??$$$$ %'F288FaK !++-K $7{F+F!N$V^tTGAq!67/09%;<45<(66QVV66QVVaZrJ   c                    t        | ||      \  }}}t        | |      }t        |dd       t        |t        | |             t	        t        | |      |d       |j                  |j                  k(  sJ |j                  |j                  dz   k(  sJ t        | t        j                  |      |      \  }}}|j                  |j                  k(  sJ |j                  |j                  dz   k(  sJ y )Nr   g+?r   r&   r   r+   )	r   rj   r   r
   r   rb   r   r0   
zeros_like)rG   rE   r   rA   rs  r   precision_recall_aucs          rH   ro  ro    s    -fgQUVAq*267C2E1=5fgF 67+-A1 66QVV66Z__q((((-g&$Aq* 66QVV66Z__q((((rJ   c           	      h   t        j                  d      5  ddg}ddg}t        |||       \  }}}t        ||      }t	        |g d       t	        |g d       t        |d       ddg}ddg}t        |||       \  }}}t        ||      }t	        |g d	       t	        |g d
       t        |d       ddg}ddg}t        |||       \  }}}t        ||      }t	        |ddg       t	        |ddg       t        |d       ddg}ddg}t        |||       \  }}}t        ||      }t	        |g d       t	        |g d       t        |d       ddg}ddg}t        |||       \  }}}t        ||      }t	        |ddg       t	        |ddg       t        |d       ddg}ddg}t        j                  t        d      5  t        |||       \  }}}d d d        t        j                  t        d      5  t        ||      }d d d        t        |g d       t        |g d       t        |d       ddg}ddg}t        |||       \  }}}t        t        ||      d       t	        |g d       t	        |g d       t        j                  ddgddgg      }t        j                  ddgddgg      }t        j                  t        d      5  t        t        ||d      d       d d d        t        j                  t        d      5  t        t        ||d      d       d d d        t        t        ||d      d       t        t        ||d      d       t        j                  ddgddgg      }t        j                  ddgddgg      }t        j                  t        d      5  t        t        ||d      d       d d d        t        j                  t        d      5  t        t        ||d      d       d d d        t        t        ||d      d       t        t        ||d      d       t        j                  ddgddgg      }t        j                  ddgddgg      }t        t        ||d      d       t        t        ||d      d       t        t        ||d      d       t        t        ||d      d       t        j                  ddgddgg      }t        j                  ddgddgg      }t        j                  t        d      5  t        t        ||d      d       d d d        t        t        ||d      d       t        j                  t        d      5  t        t        ||d      d       d d d        t        j                  t        d      5  t        t        ||d      d       d d d        t        j                  ddgddgg      }t        j                  ddgddgg      }t        t        ||d      d       t        t        ||d      d       t        t        ||d      d       t        t        ||d      d       t        j                  ddgddgg      }t        j                  ddgddgg      }t        t        ||d      d       t        t        ||d      d       t        t        ||d      d       t        t        ||d      d       d d d        t        j                  d      5  t        j                  ddgddgg      }t        j                  ddgddgg      }t        j                  t        d      5  t        t        ||d      d       d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   gxY w# 1 sw Y   8xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   y xY w)Nraise)allr   r+   r   )r|   r+   r+   r  rY   )r|   r   rY   r   r|   r   r   r   z!No positive class found in y_truer   r   )rY   rY   rY   )r+   r|   r   r   r   r   r   r   ignore)r0   errstater   r
   r   r   r   r   UserWarningr   r   )r   rG   rE   rA   rs  rt   auc_prcs          rH   #test_precision_recall_curve_toydatar    s   		! V
Qa&(DQ1a)&':!![1!!Y/GS)Qa&(DQ1a)&':!!_5!!_5 	GS)Qa&(DQ1a)&':!!c1X.!!c1X.GS)Qa&(DQ1a)&':!![1!!Y/GS)Q*(DQ1a)&':!!c1X.!!aX.GS)Q,\\+-PQ 	V,VWPTUGAq!	V\\+-PQ 	?-fg>G	?9%9%#Q,(DQ1a3FGDcJ!!_5!!]3 Aq6Aq6*+((QFQF+,\\+-PQ 	'I3	 \\+-PQ 	'Lc	 	#FGYG	
 	/QSVWAq6Aq6*+((QFQF+,\\+-PQ 	'I3	 \\+-PQ 	'Lc	 	#FGYG	
 	/QSVWAq6Aq6*+((QFQF+,#FGWEs	
 	#FGZH#	
 	#FGYG	
 	#FGWEs	
 Aq6Aq6*+((QFQF+,\\+-PQ 	'I3	 	#FGZH#	
 \\+-PQ 	'KS	 \\+-PQ 	'I3	
 Aq6Aq6*+((QFQF+,/QSVW#FGZH#	
 	#FGYG	
 	/QSVWAq6Aq6*+((S#Jc
34#FGWEs	
 	#FGZH#	
 	#FGYG	
 	#FGWEs	
iV
p 
	" Aq6Aq6*+((QFQF+,\\+-PQ 	'La		 S	V 	V	? 	? 	 		 		 		 	4	 		 		 	sV
 V
x	 		 s  F*`^#`7^'C`^4*#`_&B	`/_#`+_D`#_(<;`7_5#`3`D4`A`(0`	`(^$	`'^1	,`4^>	9`_	`_	`_%	 `(_2	-`5_?	:``	```%	!`((`1c                  D   g d} g d}t        | |d      \  }}}t        |g d       g d} g d}t        | |d      \  }}}t        |g d       g d	} g d
}t        | |d      \  }}}t        |ddg       g d} g d
}t        | |d      \  }}}t        |g d
       y)z9Check the behaviour of the `drop_intermediate` parameter.r   r   Tr   )r   r   rY   r   r   )r   r   r   r   r   rY   r   r   r   r   )r   r   r   r   r   r   r+   r+   r+   r+   N)r   r   )rG   rE   rf   rg   r   s        rH   -test_precision_recall_curve_drop_intermediater  v  s    F,G$:4%!Ivz J0 5FOG$:4%!Ivz J >? F"G$:4%!Ivz Jc
+ F"G$:4%!Ivz J 45rJ   c                      t        j                  dt              } d| d d d<   t        j                  d      }t	        | |      dk(  sJ y )Nr   rP  r+   r   r   )r0   r   r3   r   r
   rG   rE   s     rH   &test_average_precision_constant_valuesr    sE    
 XXc%FF3Q3KggclG #673t;;;rJ   c                      t        j                  ddg      } t        j                  ddg      }d}t        j                  t        |      5  t        | |d       d d d        y # 1 sw Y   y xY w)Nr   r+   z>pos_label=2 is not a valid label. It should be one of \[0, 1\]r   r&   rR   r0   r   r   r   r   r
   rG   rF   err_msgs      rH   4test_average_precision_score_binary_pos_label_errorsr    sZ    XXq!fFXXq!fFOG	z	1 =!<= = =s   A$$A-c                     t        j                  ddgddgddgddgg      } t        j                  ddgddgddgddgg      }d}t        j                  t        |      5  t        | |d	       d d d        y # 1 sw Y   y xY w)
Nr+   r   r   r   r   r   znParameter pos_label is fixed to 1 for multilabel-indicator y_true. Do not set pos_label or set pos_label to 1.r   r  r  r  s      rH   8test_average_precision_score_multilabel_pos_label_errorsr    s     XX1v1v1v1v67FXXSzC:SzC:FGF	6  
z	1 =!<= = =s    A88Bc            	          t        j                  g d      } t        j                  g dg dg dg dg dg dg      }d}t        j                  t        |      5  t        | |d	       d d d        y # 1 sw Y   y xY w)
N)r   r+   r&   r   r+   r&   )r|   r   r   )r   r|   r   )r   r   r   )r   r   r|   zdParameter pos_label is fixed to 1 for multiclass y_true. Do not set pos_label or set pos_label to 1.r   r   r  r  r  s      rH   8test_average_precision_score_multiclass_pos_label_errorsr    sq    XX()FXX	
	F	6  
z	1 =!<= = =s   A44A=c                  \   t        d      \  } }}t        | |      }t        | d|z        }t        | d|z        }t        | |dz
        }||k(  sJ ||k(  sJ ||k(  sJ t        | |      }t        | d|z        }t        | d|z        }	t        | |dz
        }
||k(  sJ ||	k(  sJ ||
k(  sJ y )NTr   r   gư>rO  )rI   r   r
   )rG   rt   rE   r   roc_auc_scaled_uproc_auc_scaled_downroc_auc_shiftedpr_aucpr_auc_scaled_uppr_auc_scaled_downpr_auc_shifteds              rH   test_score_scale_invariancer    s     )5FAwFG,G%fcGm<'w?#FGbL9O''''))))o%%%$VW5F.vsW}E0H,VWr\BN%%%%''''^###rJ   z(y_true,y_score,expected_fpr,expected_fnrr   r   )r   r   r|   )r|   r   r+   r   r|   r   r   )r|   r|   r   r   r   r   )r+   r+   r|   r   )r   r+   r+   r+   r   r  )r   r|   r|   c                 T    t        | |      \  }}}t        ||       t        ||       y Nr   r   rG   rE   expected_fprexpected_fnrrr   fnrrt   s          rH   test_det_curve_toydatar    s*    6 FG,KCaC&C&rJ   )rG   rE   r  r  r   )r+   r+   r|   r   )r+   r+   r   c                 X    t        | ||      \  }}}t        ||       t        ||       y )Nr   r  )rG   rE   r  r  r   rr   r  rt   s           rH    test_det_curve_drop_intermediater    s-    & FG?PQKCaC&C&rJ   r   r|   r|   c                 T    t        | |      \  }}}t        ||       t        ||       y r  r  r  s          rH   test_det_curve_tie_handlingr  $  s*     FG,KCaC&C&rJ   c                  T    t        t        g dg d      t        g dg d             y )Nr   r   r   )r   r   r|   r|   r+   r+   )r   r    rJ   rH   test_det_curve_sanity_checkr  8  s     )[)$&<=rJ   rE   )r   r   r|   r   r+   c                     t        g dt        j                  d|             \  }}}t        |ddg       t        |ddg       t        || t        j                  g       y )N)r   r+   r   r+   r   r+   r   r  r+   r   )r   r0   r   r   r   )rE   rr   r  rh   s       rH   test_det_curve_constant_scoresr  @  sT    #!2771g+>Ci C!Q C!Q I01rJ   rG   )r   r   r   r   r   r+   r   )r   r   r   r+   r+   r+   )r   r   r+   r+   r+   r+   )r   r+   r+   r+   r+   r+   c                 Z    t        | |       \  }}}t        |dg       t        |dg       y )Nr  r   r  )rG   rr   r  rt   s       rH   test_det_curve_perfect_scoresr  K  s.     66:KCaC!C!rJ   zy_true, y_pred, err_msgzinconsistent numbers of samplesr   r   r   z#Only one class is present in y_truer+   r+   r+   )cancerr  
not cancer)r   r   r   zpos_label is not specifiedc                 |    t        j                  t        |      5  t        | |       d d d        y # 1 sw Y   y xY w)Nr   )r   r   r   r   r  s      rH   test_det_curve_bad_inputr  \  s1      
z	1 "&&!" " "   2;c                  Z   dgdz  dgdz  z   } t        j                  g d      }d|z
  }t        | |d      \  }}}t        | |d      \  }}}|d   t        j                  d	      k(  sJ |d   t        j                  d
      k(  sJ t        ||d d d          t        ||d d d          y )Nr  r   r  r   )
r   r   r   r   r   r   r|   r   r   r   r+   r  r   r   r   rL   )r0   r   r   r   r-  r   )	rG   y_pred_pos_not_cancery_pred_pos_cancerfpr_pos_cancerfnr_pos_cancerth_pos_cancerfpr_pos_not_cancerfnr_pos_not_cancerth_pos_not_cancers	            rH   test_det_curve_pos_labelr  p  s    Z!^|nq00FHH%WX114=51NNM
 AJA=*,= v}}S1111Q6==#5555 N$6tt$<=N$6tt$<=rJ   c                    t         | ddggddgg      d       t         | ddggddgg      d       t         | ddggddgg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g d	gg dg      d       t         | g d
gg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g d	gg dg      d       t         | g d
gg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g d	gg dg      d       t         | g d
gg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | ddggddgg      d       t         | ddggddgg      d       t         | ddggddgg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g d	gg dg      d       t         | g d
gg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       t         | g dgg dg      d       y )Nr   r+   r   r   r|   r   r  r   r   r   gUUUUUU?r   g?r  g?r   r|   r   r  r|   r   r   r  )r|   r|   r|   UUUUUU?)r+   r+   r+   r   )r|   r|   r|   r|   )r   
lrap_scores    rH   check_lrap_toyr    sp   
QF8tTl^<a@
QF8tTl^<eD
QF8tTl^<a@
I;1B0CDaH
I;1B0CDeL
I;1B0CDaH
I;1B0CDeLI;!2 346I I;!2 346I 
I;1B0CDeL
I;1B0CDeLI;!2 346I 
I;1B0CDaH
I;1B0CDoV
I;1B0CDaH
I;1B0CDaH
I;1B0CDeL
I;1B0CDaH
I;1B0CDoV
I;1B0CDeLI;!2 346I 
I;1B0CDaH
I;1B0CDaH 
QF8sCj\:C@
QF8sCj\:C@
QF8sCj\:A>
I;1A0BCSI
I;1A0BCSI
I;1A0BCQG
I;1A0BCUKI;!1 235H I;!1 235H 
I;1A0BCQG
I;0ABEJ
L>4H3IJERrJ   c           	         t        d      }t        dd      D ]  }|j                  d|f      }t        j                  |      }t        j
                  d|f      } | ||      dk(  sJ  | ||      dk(  sJ t        j                  d|f      } | ||      dk(  sJ  | ||      dk(  rJ  t         | dgdgdgdggdgdgdgdgg      d       y )Nr   r&   r   r+   r   rY   r|   )r$   r[   r   r0   rx  r   r   r   )r  r*   n_labelsrE   y_score_tiesrG   s         rH   !check_zero_or_all_relevant_labelsr    s    %a(L!QK 7&&QM&:}}W- 1h-(&'*c111&,/3666 !X'&'*c111&,/36667 QC!qcA3'3%#u)EFrJ   c                    t        j                  t              5   | g dg d       d d d        t        j                  t              5   | g dg dg dg dg       d d d        t        j                  t              5   | g dg dg dg dg       d d d        t        j                  t              5   | ddgddggddg       d d d        t        j                  t              5   | ddgddggddgg       d d d        t        j                  t              5   | ddgddggdgdgg       d d d        t        j                  t              5   | ddggddgddgg       d d d        t        j                  t              5   | dgdggddgddgg       d d d        t        j                  t              5   | ddgddggdgdgg       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   _xY w# 1 sw Y   9xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   y xY w)	Nr   )r   r   r   r   )r   r   r   )r   r   r   )r   r   r   r   r+   )r   r   r   r  s    rH   check_lrap_error_raisedr    s   	z	" 09./0	z	" U90/?STU	z	" 
//R	

 
z	" -QFQF#aV,-	z	" /QFQF#q!fX./	z	" 1QFQF#qcA3Z01	z	" /QF8q!fq!f-./	z	" 1QC!:AA/01	z	" 1QFQF#qcA3Z01 1)0 0U U
 
- -/ /1 1/ /1 11 1sk   G	G$?G15G>(HHH$H0:H<G!$G.1G;>HHH!$H-0H9<Ic           	         t        dd      D ]v  }t        j                  d|f      }t        d|      D ]N  }t        ||z
        D ];  }t        j                  d|f      }d|d|||z   f<   t	         | ||      ||z         = P x y )Nr&   rO  r+   r   )r[   r0   r   r   r   )r  r  rE   
n_relevantrS   rG   s         rH   check_lrap_only_tiesr    s     !RL 	X''1h-(  8, 	XJX
23 X1h-045q#j 0001#Jvw$?hAVWX	X		XrJ   c                    t        dd      D ]  }|t        j                  |      j                  d|f      dz   z
  }t        j                  d|f      }d|d<   d|d<   t         | ||      d|z  dz   dz         t        d|      D ]g  t        |z
        D ]T  t        j                  d|f      }d|dz   f<   t         | ||      t        fdt              D                     V i  y )Nr&   rO  r+   )r   r   )r   rL   r   c              3   @   K   | ]  }|d z   |z   d z   z  z    yw)r+   Nr  ).0rs  r  rS   s     rH   	<genexpr>z>check_lrap_without_tie_and_increasing_score.<locals>.<genexpr>  s/       QC!GaK:#=>s   )r[   r0   r1   rN   r   r   rO   )r  r  rE   rG   r  rS   s       @@rH   +check_lrap_without_tie_and_increasing_scorer    s    !RL bii1991h-H1LM 1h-(tuJvw7!h,:Ja9OP  8, 	JX
23 	1h-045q#j 0001#vw/ !&z!2 		rJ   c                    t        | |       t        |       } t        |      }| j                  \  }}t        j                  |f      }t        |      D ]  }t        j                  ||   d      \  }}|j                  }||z
  t        j                  |dz         j                         }	|	   | |   j                         d   }
|
j                  dk(  s|
j                  |k(  rd||<   d||<   |
D ]*  t        fd|
D              }||xx   |   z  z  cc<   , ||xx   |
j                  z  cc<    |j                         S )z8Simple implementation of label ranking average precisionT)return_inverser+   )	minlengthr   r   c              3   4   K   | ]  }|      k    y wr  r  )r  rs  labelranks     rH   r  z_my_lrap.<locals>.<genexpr><  s      JAaDK!7 Js   )r#   r"   r/   r0   emptyr[   rM   r   bincountcumsumnonzerorO   rp  )rG   rE   r?   r  r^   r_   unique_rankinv_rankn_ranks	corr_rankrelevantn_ranked_abover  r  s               @@rH   _my_lrapr     sJ   FG, F'"G ,,IxHHi\"E9 " !#		'!*T JX""! KK!<CCE	!9$$&q)==A(!:E!Ha 	5E ! J JJN !He44H	5 	aHMM!5"8 ::<rJ   r   c                    t        dd|||      \  }}t        |j                  d   |j                  d   |      }t        |d      r|j	                         }t        ||      }t        ||      }t        ||       t        |      }|j                  ||f      }t        ||      }t        ||      }t        ||       y )Nr+   F)r@   allow_unlabeledr*   	n_classesr?   r   )n_componentsr@   r*   toarrayr   )
r   r   r/   hasattrr  r   r  r   r$   r   )	r  r  r?   r*   rt   rG   rE   
score_lrapscore_my_lraps	            rH   %check_alternative_lrap_implementationr  F  s     /!IAv $\\!_<<?!G w	"//#6vwGJVW-M
M2 &l3L""I(>"?G6vwGJVW-M
M2rJ   checkfuncc                      | |       y r  r  )r  r  s     rH   test_label_ranking_avpr  f  s     
$KrJ   c                  "    t        t               y r  )r  r   r  rJ   rH   test_lrap_error_raisedr  t  s    ABrJ   r?   )r+   r&   r   rI  r  )r&   r   rO  r*   c                 (    t        t        || |       y r  )r  r   )r?   r  r*   s      rH   $test_alternative_lrap_implementationr  x  s     *-y)\rJ   c                  l   t        j                  g dg dg dgt              } t        j                  g dg dg dg      }t        j                  g d      }t        j                  g d	      }t        t	        | ||
      t        j
                  ||z        t        j
                  |      z         y )Nr!  rm  r  rP  )r   r   r   r   )r   r   r   r   r   r   r   r   )r|   r   rY   )rY   rY   r   r   )r0   r   boolr   r   rO   )rG   rE   samplewise_lrapsr   s       rH   &test_lrap_sample_weighting_zero_labelsr    s     XX|\<@MFhh	35IJG xx 01HH_-M-G=	
 	}//0266-3HH	rJ   c            	         t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	gg dg      d       t        t        g d
gg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	gg dg      d       t        t        g d
gg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	gg dg      d       t        t        g d
gg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	g dgg dg dg      d       t        t        g d	g dg d
gg dg dg dg      d       t        t        g d	g dg d
gg dg dg dg      d       y )Nr   r+   r   r   r&   r  r  r   r   r   r   r   r   r  r  r  r  r   g      $@r   r+   r   r3  r   rO  r   rB  g@r   r+   r   r   r   r  rJ   rH   test_coverage_errorr    s   Ax4,@!DAx4,@!DAx4,@!DAx4,@!D	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L	{5F4GH!L 	9-/KL
 	9-y)/T	
 		 	9-y)/T	
 		rJ   c                     t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	gg dg      d       t        t        g d
gg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       y )Nr   r|   r+   r&   r  r  r   r   r   r   r   r   r  r  r  r  rJ   rH   test_coverage_tie_handlingr    s.   Ax3*>BAx3*>BAx3*>BAx3*>B	{5E4FGK	{5E4FGK	{5E4FGK	{5E4FGK	{5E4FGK	{5E4FGK	{5E4FGK	{5E4FGKrJ   zy_true, y_scorec                 |    t        j                  t        d      5  t        | |       d d d        y # 1 sw Y   y xY w)Nz'Expected 2D array, got 1D array insteadr   )r   r   r   r   r  s     rH   test_coverage_1d_error_messager	    s2     
z)S	T (vw'( ( (r  c            	      h   t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	gg dg      d       t        t        g d
gg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dg dgg dg dg      d       t        t        g dg dg d	gg dg dg dg      d       t        t        g dg dg d	gg dg dg dg      d       y )Nr   r+   r   r   r   r  r   r|   r   r   rY   r   r  r  r  r  r  r  r  r  rB  r  r   r   r  rJ   rH   test_label_ranking_lossr    s   *QF8tTl^DaH*QF8tTl^DaH*I;9J8KLaP*I;9J8KLeT*I;9J8KLaP*I;9J8KLeT*I;9J8KLeT*I;9J8KLeT *QF8tTl^DaH*QF8tTl^DaH*QF8sCj\BAF*QF8sCj\BAF*I;9J8KLaP*I;9J8KLaP*I;9I8JKQO*I;9I8JKQO Iy1OY3OP
 	9-y)/T	
 	"	 	9-y)/T	
 	"	rJ   csr_containerc                 z    t        t         | t        j                  g dg dg            g dg dg      d       y )Nr   r  r  r  r|   )r   r   r0   r   )r  s    rH   test_label_ranking_loss_sparser    s4    "((Iy#9:;mY=W	
 		rJ   c                  .   t        j                  t              5  t        ddgddggddg       d d d        t        j                  t              5  t        ddgddggddgg       d d d        t        j                  t              5  t        ddgddggdgdgg       d d d        t        j                  t              5  t        ddggddgddgg       d d d        t        j                  t              5  t        dgdggddgddgg       d d d        t        j                  t              5  t        ddgddggdgdgg       d d d        y # 1 sw Y   !xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   sxY w# 1 sw Y   y xY w)Nr   r+   )r   r   r   r   r  rJ   rH   $test_ranking_appropriate_input_shaper  !  s   	z	" 5QFQF+aV45	z	" 7QFQF+q!fX67	z	" 9QFQF+qcA3Z89	z	" 7QF8q!fq!f%567	z	" 9QC!:AA'789	z	" 9QFQF+qcA3Z89 95 57 79 97 79 99 9sG   EEE'?E36E?.FEE$'E03E<?FFc                     t        t        ddggddgg      d       t        t        ddggddgg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g dgg dg      d       t        t        g d	gg dg      d       t        t        g d
gg dg      d       y )Nr+   r   r|   r   r  r   r   r   r   r  r  r  rJ   rH   test_ranking_loss_ties_handlingr  1  s    *QF8sCj\BAF*QF8sCj\BAF*I;9I8JKUS*I;9I8JKUS*I;9I8JKQO*I;9I8JKQO*I;9I8JKQO*I;9I8JKQOrJ   c                      t        dd      \  } }| dz   }t        ||       t        j                  j	                  d      j                  d      \  }}t        ||       y Nr   rO  )r*   r  r+   )r&   r   rO  )r   _test_dcg_score_forr0   r4   r5   random_samplert   rG   rE   s      rH   test_dcg_scorer  =  sT    .ALIAvgkG(ii++A.<<\JOFG(rJ   c                     t        j                  t        j                  | j                  d         dz         }t	        | |       }t	        | |      }||k  j                         sJ t	        | | d      |k  j                         sJ |j                  | j                  d   fk(  sJ |j                  | j                  d   fk(  sJ |t        j                  t        j                  |       d d d d df   |z  j                  d            k(  sJ y )Nr+   r&   r   kr   rL   axis)
r0   log2r1   r/   r   r|  r   r-  sortrO   )rG   rE   discountidealr^   s        rH   r  r  E  s    wwryya1A56Hvv.Evw/EUN!!!vv3u<AACCC;;6<<?,,,,;;6<<?,,,,FMM2776?1dd7#;h#F"K"KQR"K"STTTTrJ   c            	         t        j                  t        j                  d      g      } t        j                  | j                        }t        | |      }t        | |d      }dt        j                  t        j                  dd            z  }|t        j                  |j                         | j                         z  g      k(  sJ |t        j                  || d d d d df   z  j                         g      k(  sJ d|dd	d f<   t        | |      }t        | |d      }|t        j                  || d d d d df   z  j                         g      k(  sJ |t        j                  |d d j                         | dd	d f   j                         z  |dd  j                         | dd d	f   j                         z  z   g      k(  sJ y )
Nr   Tignore_tiesr+   r&   r   rL   r   r   )r0   r9  r1   r   r/   r   r  r   r-  rO   rp  )rG   rE   dcgdcg_ignore_ties	discountss        rH   test_dcg_tiesr)  P  s   ZZ1'Fhhv||$G
VW
-C(dKOBGGBIIaO,,I&--6;;=!@ ABBBBfmmi&DbD/.I-N-N-P,QRRRRGAqrEN
VW
-C(dKOfmmi&DbD/.I-N-N-P,QRRRR&--bqM&AB-"4"4"66m!F1bqb5M$6$6$889	
   rJ   c            	          t        j                  d      j                  d      } t        | | dd      t	        j
                  t        | | dd            k(  sJ y )N   )r&   r   r   T)r  r%  )r0   r1   rN   r   r   r-  )r   s    rH   test_ndcg_ignore_ties_with_kr,  d  sP    
		"f%AaaT2fmm1a1$/7   rJ   c                      t        j                  g dg      } t        j                  g dg      }d}t        j                  t        |      5  t        | |       ddd       y# 1 sw Y   yxY w)zDCheck `ndcg_score` exception when `y_true` contains negative values.)g{Gzg(\gGz޿g(\?gQ?)gQ?gףp=
?r   gQ?gHzG?z7ndcg_score should not be used on negative y_true valuesr   N)r0   r   r   r   r   r   )rG   rE   r   s      rH    test_ndcg_negative_ndarray_errorr.  k  sZ    XX89:Fhh678GP	z)9	: $67#$ $ $s   A$$A-c                     t        j                  d      j                  dd      } | t         j                  j	                  d      j                  dd| j                        z   }t        | |      }t        | |d	      }|t        j                  |      k(  sJ |t        j                  d
      k(  sJ |dz  }t        | |      t        j                  d
      k(  sJ y )NF   r   rO  r   皙ɿr   r   Tr$  rY   r4  )
r0   r1   rN   r4   r5   r   r/   r   r   r-  )rG   rE   ndcgndcg_no_tiess       rH   test_ndcg_invariantr4  t  s    YYr]""1b)Fryy,,Q/77c7UUGfg&Dfg4@L6==....6==%%%%tOGfg&&--*<<<<rJ   r%  c           
      ,   dt        j                  d      d d z  }t        j                  t        j                  ddd      d      }|t         j                  j                  d      j                  dd	|j                  
      z   }t        |||       t        j                  dt        j                  t        j                  dd            z        k(  sJ t        |||       t        j                  dt        j                  t        j                  dd            z        k(  sJ t        |||       t        j                  dt        j                  t        j                  dd            z        k(  sJ t        ||d|       t        j                  dt        j                  t        j                  dd            z        k(  sJ t        |||       t        j                  dt        j                  t        j                  dd            z  j                               k(  sJ t!        |||       t        j                  dt        j                  t        j                  dd            z  j                               k(  sJ dt        j"                  d      z  }dt        j                  t        j                  dd            z  j%                         }t        |||       t        j                  |t        j"                  d      z        k(  sJ t        |||       t        j                  t        j"                  d            k(  sJ t!        |||       t        j                  |      k(  sJ t        |||       t        j                  d      k(  sJ y )Nr   r   r   r   rL   )r   r+   r   r1  r   r   r$  r&   r+   rO  )log_baser%  )r   r   	   rY   )r0   eyetiler1   r4   r5   r   r/   r   r   r-  r  r   log10r   rp  r   r   rO   )r%  rG   rE   y_score_noisyexpected_dcg_scores        rH   test_ndcg_toy_examplesr=    s   2AFggbii2r*F3Gbii33A6>>c ?  M [	q277299Q?33	45 5 5 ;	q277299Q?33	45 5 5 [	q277299Q?33	45 5 5 "+	q288BIIaO44	56 6 6 fg;?6==	
RWWRYYq!_%	%++-D    VW+>&--	
RWWRYYq!_%	%++-C     Fbggbii1o66;;=[	)BGGAJ6	78 8 8 [	rwwqz	"# # # VW+>&--C    fg;?6==QTCUUUUrJ   c                      d} t        j                  t        |       5  t        dggdgg       ddd       y# 1 sw Y   yxY w)zhCheck that we raise an informative error message when trying to
    compute NDCG with a single document.zTComputing NDCG is only meaningful when there is more than 1 document. Got 1 instead.r   r+   N)r   r   r   r   )r  s    rH   test_ndcg_error_single_documentr?    sC    	  
z	1 !QC5A3% ! ! !s	   8Ac                      t        dd      \  } }| dz   }t        ||       t        j                  j	                  d      j                  d      \  }}t        ||       y r  )r   _test_ndcg_score_forr0   r4   r5   r  r  s      rH   test_ndcg_scorerB    sT    .ALIAvgkG)ii++A.<<\JOFG)rJ   c                    t        | |       }t        | |      }||k  j                         sJ | dk(  j                  d      }||    t        j                  t	        j
                  | j                                     k(  sJ ||   t        j                  t	        j                  |j                                     k(  sJ ||    t        j                  t        | |      |    t        | |       |    z        k(  sJ ||   t        j                  t	        j                  |j                                     k(  sJ |j                  | j                  d   fk(  sJ |j                  | j                  d   fk(  sJ y )Nr   r+   r  )
r   r|  r   r-  r0   r   rO   r   r   r/   )rG   rE   r"  r^   all_zeros        rH   rA  rA    sR   /E0EUN!!!!  a (H(v}}RWWxi__5F-GHHHH?fmmBHHX\\^,DEEEE(v}}67+XI6
VV
,hY
7	8     ?fmmBHHX\\^,DEEEE;;6<<?,,,,;;6<<?,,,,rJ   c            	      .   t        j                  g d      } t        | | d      dk(  sJ t        | | d      dk(  sJ t        j                  t
              5  t        | | d      sJ 	 d d d        t        j                  t
              5  t        | | d      sJ 	 d d d        t        j                  t
              5  t        | | d      sJ 	 d d d        t        j                  g d      }t        | |d      }t        | |      }||k(  sJ t        | |d	      d
k(  sJ t        d      \  } }}t        j                  ddd      D ]%  }t        t        | ||      t        | ||             ' y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   xY w)Nr"  r+   )rq   gMbP?gg?r   )r   r   r   g{Gz?r   r|   Tr   g-C6?r   )
r0   r   r   r   r   r   rI   linspacer   r   )rG   r	  roc_auc_with_max_fpr_oneunconstrained_roc_aucrF   rt   rq   s          rH   test_partial_roc_auc_scorerI    sw   XXl#F3q88871<<<	z	" ;VVT:::;	z	" :VVS999:	z	" 8VVQ7778 xx+,H,VXqI)&(;#'<<<<373>>>'t4FFA;;tQ* 
&&':"667;	

; ;: :8 8s$   E2
E?=F2E<?FFzy_true, k, true_scorer   r&   r   r   c                     t        j                  g dg dg dg dg      }t        | ||      }|t        j                  |      k(  sJ y )Nr  r   r   r   r   r   r   r   r   r   r   r   r   r  r0   r   r   r   r-  )rG   r  
true_scorerE   r^   s        rH   test_top_k_accuracy_scorerP    sJ     hh    		
G !A6EFMM*----rJ   zy_score, k, true_score)rL   rL   r+   r+   )rL   r+   rL   r+   )r   r   r   r   )r   r   r   r   c                 ,   g d}| j                         dk\  r| j                         dk  rdnd}|dk(  r"| |kD  j                  t        j                        n|}t        || |      }t        ||      }||cxk(  rt        j                  |      k(  sJ  J y )Nr"  r   r+   r|   r  )	minmaxastyper0   int64r   r   r   r-  )rE   r  rO  rG   rh   rF   r^   	score_accs           rH    test_top_k_accuracy_score_binaryrW    s     F{{})gkkmq.@aI78Avg	!))"((36F A6Evv.II:z!::::::rJ   zy_true, true_score, labels)r   r+   r+   r&   r  )r   rC  rC  r   )r   r   r  rC  labels_as_ndarrayc                     |rt        j                  |      }t        j                  g dg dg dg dg      }t        | |d|      }|t	        j
                  |      k(  sJ y)z,Test when labels and y_score are multiclass.r  rK  rL  rM  r&   r  r  N)r0   r9  r   r   r   r-  )rG   rO  r  rX  rE   r^   s         rH   0test_top_k_accuracy_score_multiclass_with_labelsr[    s]     F#hh    		
G !AfEEFMM*----rJ   c            
         t        j                  dddd      \  } }t        | |d      \  }}}}t        d      }|j	                  ||       t        ||f||f      D ]f  \  } }t        dd      D cg c]  }t        ||j                  |       |      ! }}t        j                  t        j                  |      dkD        rfJ  y c c}w )NrO  r4  r   )r  r?   n_informativer*   r)   r&   r  )r   make_classificationr   r   r8   rY  r[   r   r9   r0   r|  r   )	r=   r>   X_trainX_testy_trainy_testrD   r  scoress	            rH   $test_top_k_accuracy_score_increasingrd  ,  s    ''BQDAq (811'M$GVWf
!
,CGGGWWf%'89 +1HMaQS
CD C$5$5a$8A>
 
 vvbggfo)***+
s   2$Cc                     t        j                  g dg dg dg dg      }t        | ||      t        j                  |      k(  sJ y )N)r   r   r   r   )r+   r   r   r   )r   r   r   r   r  r  rN  )rG   r  rO  rE   s       rH   test_top_k_accuracy_score_tiesrf  ?  sE     hh		
G  15z9RRRRrJ   z	y_true, kr   c                     t        j                  g dg dg dg dg      }d}t        j                  t        |      5  t        | ||      }d d d        dk(  sJ y # 1 sw Y   xY w)	Nr  )r   r   r   r   )r   r   r   r   )r   r   r   r   zu'k' \(\d+\) greater than or equal to 'n_classes' \(\d+\) will result in a perfect score and is therefore meaningless.r   r  r+   )r0   r   r   r   r   r   )rG   r  rE   r   r^   s        rH   !test_top_k_accuracy_score_warningrh  T  sn     hh    		
G	6  
,4D	E ;$VW:;A::; ;s   AA'zy_true, y_score, labels, msg)r   g=
ףp=?r+   r&   )r   r   r   )r   r   r   r   )r   r|   r   z9y type must be 'binary' or 'multiclass', got 'continuous'zZNumber of classes in 'y_true' \(4\) not equal to the number of classes in 'y_score' \(3\).)r   r   r   r   z"Parameter 'labels' must be unique.z#Parameter 'labels' must be ordered.)r   r   r+   r&   zSNumber of given labels \(4\) not equal to the number of classes in 'y_score' \(3\).r  z3'y_true' contains labels not in parameter 'labels'.)r|   r   r   )r   r   r   z}`y_true` is binary while y_score is 2d with 3 classes. If `y_true` does not contain all the labels, `labels` must be providedc                     t        j                  t        |      5  t        | |d|       d d d        y # 1 sw Y   y xY w)Nr   r&   rZ  )r   r   r   r   )rG   rE   r  rE  s       rH   test_top_k_accuracy_score_errorrj  m  s:    f 
z	- BVW&AB B Bs   5>c                      | g dg dg      }t        j                  g dg dg      }t        ||      }|t        j                  d      k(  sJ y )Nr   r   )r|   r   r   r  )r0   r   r   r   r-  )r  rG   rE   results       rH   Otest_label_ranking_avg_precision_score_should_allow_csr_matrix_for_y_true_inputrm    sI     Iy12Fhh34G267CFV]]5))))rJ   metricr6  ))FT)r   r+   )r   rY   )zeroonec                    t         j                  j                  d      }d|d   }}|j                  ||d      }|j	                  |      } | |||      }t        |t              rt        j                  |      rJ y|\  }}	}
t        j                  |      j                         rJ t        j                  |	      j                         rJ t        j                  |
      j                         rJ y)zCheck that the metric works with different types of `pos_label`.

    We can expect `pos_label` to be a bool, an integer, a float, a string.
    No error should be raised for those types.
    *   rO  rL   T)r   replacer  N)	r0   r4   r5   choicerJ  
isinstancerP   r   any)rn  r6  rB   r?   rR   rG   y_probarl  metric_1metric_2r   s              rH   #test_ranking_metric_pos_label_typesrz    s     ))


#Cwr{yIZZiZ>Fhhy!GFGy9F&% 88F####)/&(J88H%))+++88H%))+++88J'++----rJ   c                     t         j                  j                  |       }|j                  ddd      }|j	                  d      }t        ||      \  }}}t        j                  |d         sJ y)zCheck that thresholds do not exceed 1.0 when `y_score` is a probability
    estimate.

    Non-regression test for:
    https://github.com/scikit-learn/scikit-learn/issues/26193
    r   r&   rO  r   N)r0   r4   r5   r   rJ  r   isinf)r   rB   rG   rE   rt   r   s         rH   (test_roc_curve_with_probablity_estimatesr}    sc     ))

 2
3C[[AB['FhhrlG 1Aq*88JqM"""rJ   )NF)r   rI  r   )r   r   numpyr0   r   scipyr   sklearnr   sklearn.datasetsr   sklearn.exceptionsr   sklearn.linear_modelr   sklearn.metricsr   r	   r
   r   r   r   r   r   r   r   r   r   r   r   sklearn.metrics._rankingr   r   sklearn.model_selectionr   sklearn.preprocessingr   sklearn.random_projectionr   sklearn.utils._testingr   r   r   r   r   sklearn.utils.extmathr    sklearn.utils.fixesr!   sklearn.utils.validationr"   r#   r$   CURVE_FUNCSrI   rW   rb   rj   r   r   markparametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r)  r1  rA  rF  rM  rS  rV  r_  filterwarningsre  rk  rv  ro  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r[   r  r  r  r  r	  r  r  r  r  r  r  r)  r,  r.  r4  r=  r?  rB  rA  rI  rP  rW  r[  rd  rf  rh  rj  rm  rz  r}  r  rJ   rH   <module>r     s    	     ; 5 3     M 4 0 ;  * .  #	)#`2<&H86& $/
) 0
)
")&#))8)6oNdM	).&" 	,	+	,	&	/	t$	/X/Xd 	,	+	&	'9: 	,	&	t$	y!	/	D D:( 	,hbhh|.DiP/BHH)*	
 BHH\"	
 1BHH)*	
 BHH\"F	
 BHH)*#J	
 BHH\"	
 BHH)* 	
 ABHH)*	
 ABHH)*	
 ABHH\"	
}CFN 7P 8OFPP ' "%8	
' "%8	
F "B7	
 "c2	
 
4R8="F	0G"F	0&* {3# 4# {3E 4E0 LM(>	'JK(9:' ; L N' {3	& 4	& $/  0 >), $/` 0`F6D<=
==*$2 .	K!qc*	NQC!-	NQC!-	%sQC0	K	:	NM9=	NM9=	%}i@	K#,	NQC!-	NQC!-	%sQC0	K>	NNLA	NNLA	%~|D	KM:	NI}=	NI}=	%y-@)2'32' N 
K!1<G	Ki>	N$4lEJ	NKDA	KM5A	KM4@	NI}eD	NI}dC'' .
Q#saVaV,
Q#saVaV,	$sAhA7	$sAhA7	$qcA3/	$i;	$i=	$i=	'' $EF2 G2 							 
Q?@	QH?@	K!FG	K!FG.(	

"">48Sv.14X6#N 9:3@ 3)	 "G!RS TC m4j1q2 3 2 5*4nL  	$%	%&'
&'(((V .9 :9 	P)U($= u6#V 7#VL!*- 
4 	q$	q#	q$
.
. 	.	!1a(	.	!1c*	.	!1a(	&	'A.	&	'C0	&	'A.
	;
	;  	,	|4	,	l3	,	l3	&	'/CD	 ,tUm<. =.&+& 	q$	q#	q!
S
S 	q	q$ " 	 G
	
 	 +	
 !	 !0
	
 !	 1
	
 	  	
 	 A
	
 Fo.W	
KNQdBeQdB
 .9* :* &	3I9U C..(#rJ   