
    -iH                       S SK r S SKrS SKrS SKrS SKJr  S SKJrJ	r	  S SK
Jr  S SKJr  S SKJr  S SKJrJrJrJrJrJrJrJrJrJrJrJrJr  S SKJrJ r   S S	K!J"r"  S S
K#J$r$  S SK%J&r&  S SK'J(r(J)r)J*r*J+r+J,r,  S SK-J.r.  S SK/J0r0  S SK1J2r2J3r3J4r4  \\\/r5SS jr6S r7S r8S r9S r:\Rv                  Ry                  SSS/5      S 5       r=S r>S r?S r@S rAS rBS rCS rDS  rES! rFS" rGS# rH\Rv                  Ry                  S$\R                  " / S%Q5      / S&Q4\R                  " / S%Q5      S4/ S'Q/ S(Q4/ S'QS4/5      S) 5       rJ\Rv                  Ry                  S$\R                  " / S*Q5      / S&Q4\R                  " / S+Q5      / S,Q4/5      S- 5       rK\Rv                  Ry                  S$\R                  " / S.Q5      S4/ S/QS4/ S.Q/ S&Q4/ S/Q/ S(Q4/5      S0 5       rL\Rv                  Ry                  S1/ S2Q5      S3 5       rMS4 rN\Rv                  Ry                  S5S6\R                  " / S.Q5      / S7Q4S6\R                  " / S/Q5      / S8Q4S9\R                  " / S*Q5      S4S:\R                  " / S/Q5      / S;Q4S<\R                  " / S.Q5      S S=/4S<\R                  " / S/Q5      S>S?/4S@\R                  " / S.Q5      / SAQ4S@\R                  " / S/Q5      / SBQ4SC\R                  " / SDQ5      / S(Q4SC\R                  " / SBQ5      / S(Q4SC\R                  " / SAQ5      / S&Q4/5      \Rv                  Ry                  SESFSG/5      SH 5       5       rO\Rv                  Ry                  SISJSKSFSL.4SMSKSGSL.4SNSF/ SO.4SPSFSQSR.4SS0 4/5      ST 5       rPSU rQ\Rv                  Ry                  SV\55      SW 5       rR\Rv                  Ry                  SV\55      SX 5       rS\Rv                  R                  SY5      \Rv                  Ry                  SV\\/5      \Rv                  Ry                  SZS[S\/5      S] 5       5       5       rU\Rv                  Ry                  SV\55      S^ 5       rV\Rv                  Ry                  SSS/5      S_ 5       rWS` rX\Rv                  Ry                  SSS/5      Sa 5       rYSb rZSc r[Sd r\Se r]Sf r^Sg r_\Rv                  Ry                  Sh/ SiQ/ SjQS /S /4/ SiQ/ SkQS /S /4/ SiQ/ SlQS /S /4/ SiQ/ SmQS /S /4/ SnQ/ SjQ/ SoQ/ SpQ4/ SnQ/ SkQ/ SoQ/ SpQ4/ SnQ/ SlQ/ SoQ/ SpQ4/ SnQ/ SmQ/ SoQ/ SpQ4/ SpQ/ SjQSq/S /4/ SpQ/ SkQS /S /4/ SpQ/ SlQS /S /4/ SpQ/ SmQS /S /4/ SrQ/ SjQ/ SsQ/ StQ4/ SrQ/ SkQ/ SsQ/ StQ4/ SrQ/ SlQ/ SsQ/ StQ4/ SrQ/ SmQ/ SsQ/ StQ4/ SuQ/ SjQ/ SvQ/ SwQ4/ SuQ/ SkQ/ SvQ/ SwQ4/ SuQ/ SlQ/ SvQ/ SwQ4/ SuQ/ SmQ/ SvQ/ SwQ4/5      Sx 5       r`\Rv                  Ry                  / SyQ/ SrQ/ SjQ/ SzQ/ StQS4/ SrQ/ SjQ/ S{Q/ SpQS4/ SrQ/ SkQ/ SzQ/ StQS4/ SrQ/ SkQ/ S{Q/ SpQS4/ SuQ/ SjQ/ SvQ/ SwQS4/ SuQ/ SjQ/ SvQ/ SwQS4/ SuQ/ SkQ/ SvQ/ SwQS4/ SuQ/ SkQ/ SvQ/ SwQS4/5      S| 5       ra\Rv                  Ry                  ShS=S /SQSQ/S=S /S S=/4S S=/SQSQ/S=S /S S=/4/ SiQ/ S}QSQS /S S=/4/ SnQ/ S}QSQS /S S=/4/ SpQ/ S}QS /S /4/ SrQ/ S}Q/ SvQ/ SpQ4/ SuQ/ S}Q/ SvQ/ SjQ4/ SvQ/ S}Q/ SvQ/ SjQ4/5      S~ 5       rbS rc\Rv                  Ry                  S/ SQ5      S 5       rd\Rv                  Ry                  S/ SQ/ SQ/ SQ/ SQ/ SQ/5      S 5       re\Rv                  Ry                  SS S=// SjQS4/ SpQS SQ/S4/ SQ/ SjQS4/ SQ/ SjQS4/ SQ/ SQS4/5      S 5       rfS rgS rhS riS rjS rkS rlS rm SS jrn\Rv                  Ry                  S\h\l\k\i45      \Rv                  Ry                  S\\m45      S 5       5       roS rp\Rv                  Ry                  SS5      \Rv                  Ry                  SS5      \Rv                  Ry                  S\q" S=5      5      S 5       5       5       rrS rsS rtS ru\Rv                  Ry                  S/ SuQ/ S}Q4/ SuQ/ S}Q/4/ SuQ// S}Q4/5      S 5       rvS rw\Rv                  Ry                  S\05      S 5       rxS ryS rzS r{S r|S r}S r~S rS r\Rv                  Ry                  SSS/5      S 5       rS rS rS rS r\Rv                  Ry                  S/ SAQS=S4/ SAQSSQ4/ SAQSS4/5      S 5       r\Rv                  Ry                  S\R                  " / SQ5      S=S=4\R                  " / SQ5      S=SQ4\R                  " / SQ5      SS=4\R                  " / SQ5      S=S=4\R                  " / SQ5      S=SQ4\R                  " / SQ5      SS=4/5      S 5       r\Rv                  Ry                  S\R                  " / SQ5      S/ SAQ4\R                  " / StQ5      SQ/ SAQ4\R                  " / SQ5      SQ/ SAQ4\R                  " / SQ5      S/ SQ4/5      \Rv                  Ry                  SSS/5      S 5       5       rS r\Rv                  Ry                  S/ SAQS=S4/ SAQSSQ4/ SAQSS=4/5      S 5       r\Rv                  Ry                  S/ SAQS4/ SAQS4/5      S 5       r\Rv                  Ry                  S/ SQ/ SQ/ SQ/ SQ/ SQ/SS4/ SAQ/ SQ/ SQ/ SQ/ SQ/SS4/ SQ/ SQ/ SQ/ SQ/ SQ// S/QS4/ SQ/ SQ/ SQ/ SQ/ SQ// S;QS4/ SQ/ SQ/ SQ/ SQ/ SQ// SAQS4/ SQ/ SQ/ SQ/ SQ/ SQ// SQS4S S=// SQ/ SQ/SS4/5      S 5       r\Rv                  Ry                  S\05      S 5       r\Rv                  Ry                  S\\\\/5      \Rv                  Ry                  S/ SQ5      S 5       5       rS rg)    N)stats)datasetssvm)make_multilabel_classification)UndefinedMetricWarning)LogisticRegression)accuracy_scoreaucaverage_precision_score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                    U c  [         R                  " 5       n U R                  nU R                  nU(       a  X#S:     X3S:     p2UR                  u  pE[
        R                  " U5      n[        S5      nUR                  U5        X&   X6   p2[        US-  5      n[
        R                  R                  S5      n[
        R                  X'R                  USU-  5      4   n[        R                  " SSSS9n	U	R!                  USU USU 5      R#                  X(S 5      n
U(       a	  U
SS2S	4   n
U	R%                  X(S 5      nX8S nXU
4$ )
zMake some classification predictions on a toy dataset using a SVC

If binary is True restrict to a binary classification problem instead of a
multiclass classification problem
N   %   r      linearT)kernelprobabilityrandom_state   )r   	load_irisdatatargetshapenparanger$   shuffleintrandomRandomStatec_randnr   SVCfitpredict_probapredict)datasetbinaryXy	n_samples
n_featuresprnghalfclfy_scorey_predy_trues                U/var/www/html/venv/lib/python3.13/site-packages/sklearn/metrics/tests/test_ranking.pymake_predictionrL   8   s5    $$&AAQxq51GGI
		)A
R
 CKKN4qy1}D ))


"C
a9cJ&6778A ''t!
DCggah%4)77%AG !Q$-[[5"FuXF7""    c                    [         R                  " U 5      S   nXU:H     nXU:g     nUR                  SS5      UR                  SS5      -
  n[         R                  " US:  5      nU[	        [        U5      [        U5      -  5      -  $ )zGAlternative implementation to check for correctness of
`roc_auc_score`.r-   r   )r2   uniquereshapesumfloatlen)rJ   rH   	pos_labelposnegdiff_matrix	n_corrects          rK   _aucrZ   h   s     		&!!$I I%
&C
I%
&C++a$s{{2q'99K{Q'IuSXC0111rM   c                 f   [         R                  " U 5      S   n[         R                  " X:H  5      n[         R                  " U5      SSS2   nX   nX   n Sn[	        [        U5      5       H?  nX   U:X  d  M  Sn[	        SUS-   5       H  nX   U:X  d  M  US-  nM     XvS-   -  nXW-  nMA     XS-  $ )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-   NrO   r         ?)r2   rP   rR   argsortrangerT   )	rJ   rH   rU   n_posorderscoreiprecjs	            rK   _average_precisionre   w   s     		&!!$IFF6&'EJJw"%EnG]FE3w< 9	! D1a!e_9	)CKD % GODME ! =rM   c                     [        X5      u  p#n[        [        U5      5      n[        [        U5      5      nSn[        S[	        U5      5       H  nXRU   X6   X6S-
     -
  -  -  nM     U$ )aS  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^   rT   )rJ   rH   	precisionrecall	thresholdaverage_precisionrb   s          rK   _average_precision_slowrm      sq     $:&#J IyXi()I(6"#F1c)n%q\VYA-FGG &rM   c                 z    S nU" XU5      u  pE[        XE5      nSnUnSX-
  -  X-   -  n	X-
  n
SSXi-
  X-
  -  -   -  $ )z[Alternative implementation to check for correctness of `roc_auc_score`
with `max_fpr` set.
c                    [        X5      u  p4nX3U:*     n[        R                  " Xb5      nXCU:*     n[        R                  " X2:  5      nUS-
  n	X9   X8   /n
XI   XH   /n[        R                  " U[        R                  " X*U5      5      nXg4$ )Nr-   )r   r2   appendargmaxinterp)rJ   	y_predictmax_fprfprtpr_new_fprnew_tpridx_outidx_inx_interpy_interps               rK   _partial_roc,_partial_roc_auc_score.<locals>._partial_roc   s    2!Wn%))G-Wn%))CM*1K.K.))GRYYw(%KL!!rM   r         ?r-   )r
   )rJ   rs   rt   r~   rx   ry   partial_aucfpr1fpr2min_areamax_areas              rK   _partial_roc_auc_scorer      sc    

" $Fw?Gg'K DDdk"dk2H{H!{-(2EFFGGrM   dropTc                    [        SS9u  pn[        X5      n[        XU S9u  pVn[        XV5      n[	        XSS9  [        U[        X5      5        UR                  UR                  :X  d   eUR                  UR                  :X  d   eg )NTr?   drop_intermediater&   decimal)rL   rZ   r   r
   r   r   r   r1   )	r   rJ   rw   rH   expected_aucru   rv   
thresholdsroc_aucs	            rK   test_roc_curver      s~     )5Fw(L$VMCj#mGgQ?v!?@99		!!!99
(((((rM   c                  Z   [         R                  R                  S5      n [         R                  " S/S-  S/S-  -   5      nU R	                  SSS9n[        XSS9u  p4nUS   S:X  d   eUS	   S:X  d   eUR                  UR                  :X  d   eUR                  UR                  :X  d   eg )
Nr   2   r-      d   sizeTr   rO   )r2   r6   r7   arrayrandintr   r1   )rE   rJ   rI   ru   rv   thrs         rK   test_roc_curve_end_pointsr      s     ))


"CXXqcBh!r)*F[[[%FfEMCcq6Q;;r7a<<99		!!!99		!!!rM   c                  b   [        SS9u  pn[        X5      u  p4n/ nU HK  n[        R                  " X':  U -  5      n[        R                  " U 5      n	UR	                  SU-  U	-  5        MM     [        XFSS9  UR                  UR                  :X  d   eUR                  UR                  :X  d   eg )NTr   r\   r&   r   )rL   r   r2   rR   rp   r   r1   )
rJ   rw   rH   ru   rv   r   tpr_correctttprD   s
             rK   test_roc_returns_consistencyr      s     )5Fw$V5Cj KVVW\V+,FF6N38a<(  c:99		!!!99
(((((rM   c                      [        SS9u  pn[        R                  " [        5         [	        X5        S S S 5        g ! , (       d  f       g = f)NFr   )rL   pytestraises
ValueErrorr   )rJ   rw   rH   s      rK   test_roc_curve_multir      s1    (6Fw	z	"&" 
#	"	"s	   <
A
c                      [        SS9u  pn[        XS-
  5      u  p4n[        X45      n[        USSS9  UR                  UR                  :X  d   eUR                  UR                  :X  d   eg )NTr   r   ?r&   r   )rL   r   r
   r   r1   )rJ   rw   rH   ru   rv   r   r   s          rK   test_roc_curve_confidencer      sg    (5Fw$Vs];Cj#mGgtQ799		!!!99
(((((rM   c                     [        SS9u  pn[        R                  " U R                  5      n[	        X5      u  pEn[        XE5      n[        USSS9  UR                  UR                  :X  d   eUR                  UR                  :X  d   e[        R                  " U R                  5      n[	        X5      u  pEn[        XE5      n[        USSS9  UR                  UR                  :X  d   eUR                  UR                  :X  d   e[	        X5      u  pEn[        XE5      n[        USSS9  UR                  UR                  :X  d   eUR                  UR                  :X  d   eg )NTr   r   r&   r   g(\?)rL   r2   onesr1   r   r
   r   zeros)rJ   predrH   trivial_predru   rv   r   r   s           rK   test_roc_curve_hardr     s1   +48F' 776<<(L$V:Cj#mGgtQ799		!!!99
((((( 88FLL)L$V:Cj#mGgtQ799		!!!99
((((( %V2Cj#mGgtQ799		!!!99
(((((rM   c                     / SQn / SQnSn[         R                  " [        US9   [        X5      u  p4nS S S 5        [	        W[
        R                  " [        W5      [
        R                  5      5        UR                  WR                  :X  d   eUR                  UR                  :X  d   eSn[         R                  " [        US9   [        U  Vs/ s H  nSU-
  PM
     snU5      u  p4nS S S 5        [	        U[
        R                  " [        U5      [
        R                  5      5        UR                  UR                  :X  d   eUR                  UR                  :X  d   eg ! , (       d  f       GN7= fs  snf ! , (       d  f       N= f)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   r2   fullrT   nanr1   )rJ   rI   expected_messageru   rv   r   xs          rK   test_roc_curve_one_labelr   !  s8   +F+F 	T  
,4D	E(8* 
F sBGGC
ORVV<=99		!!!99
((((( 	S  
,4D	E()@A!a%)@&I* 
F sBGGC
ORVV<=99		!!!99
(((((# 
F	E *A 
F	Es)   E#
E:E5
E:#
E25E::
Fc                  2   SS/n SS/n[        X5      u  p#n[        X5      n[        U/ SQ5        [        U/ SQ5        [        US5        SS/n SS/n[        X5      u  p#n[        X5      n[        U/ SQ5        [        U/ SQ5        [        US5        SS/n SS/n[        X5      u  p#n[        X5      n[        USS/5        [        USS/5        [        US5        SS/n SS/n[        X5      u  p#n[        X5      n[        U/ SQ5        [        U/ SQ5        [        US5        SS/n SS/n[        X5      u  p#n[        X5      n[        USS/5        [        USS/5        [        US5        SS/n SS	/nS
n[        R
                  " [        US9   [        X5      u  p#nS S S 5        [        U/ SQ5        [        U[        R                  [        R                  [        R                  /5        Sn[        R
                  " [        US9   [        X5      nS S S 5        [        R                  " W5      (       d   eSS/n SS	/nSn[        R
                  " [        US9   [        X5      u  p#nS S S 5        [        U[        R                  [        R                  [        R                  /5        [        U/ SQ5        Sn[        R
                  " [        US9   [        X5      nS S S 5        [        R                  " U5      (       d   e[        R                  " SS/SS//5      n [        R                  " SS/SS//5      n[        R
                  " [        US9   [        XSS9  S S S 5        [        R
                  " [        US9   [        XSS9  S S S 5        [        [        XSS9S5        [        [        XSS9S5        [        R                  " SS/SS//5      n [        R                  " SS/SS//5      n[        R
                  " [        US9   [        XSS9  S S S 5        [        R
                  " [        US9   [        XSS9  S S S 5        [        [        XSS9S5        [        [        XSS9S5        [        R                  " SS/SS//5      n [        R                  " SS/SS//5      n[        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [        R                  " SS/SS//5      n [        R                  " SS/SS//5      n[        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        g ! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GNG= f! , (       d  f       GN= f! , (       d  f       GNt= f! , (       d  f       GNZ= f! , (       d  f       GN= f! , (       d  f       GN= f)Nr   r-   r   r   r-   r   r-   r-   r\           r         ?      ?r   r   r   r   r\   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   r2   r   mathisnanr   )rJ   rH   rv   ru   rw   r   r   r
   s           rK   test_roc_curve_toydatar   <  s?   VF!fGF,KCaF,Gc9-c9-%VF!fGF,KCaF,Gc9-c9-%VF!fGF,KCaF,GcAq6*cAq6*%VF!fGF,KCaF,Gc9-c9-%VFCjGF,KCaF,GcAq6*cAq6*% VFTlG 	S  
,4D	E0! 
Fc?3cBFFBFFBFF#;<	5  
,4D	EF, 
F::c??? VFTlG 	T  
,4D	E0! 
FcBFFBFFBFF#;<c?3	5  
,4D	EF, 
F::c??? XX1v1v&'FhhAA'(G	,4D	Efw7 
F	,4D	Efz: 
FfyI3OfwGMXX1v1v&'FhhAA'(G	,4D	Efw7 
F	,4D	Efz: 
FfyI3OfwGMXX1v1v&'FhhAA'(GfwGKfzJANfyI1MfwGKXX1v1v&'Fhhc
S#J/0GfwGMfzJCPfyI3OfwGM} 
F	E 
F	E 
F	E 
F	E 
F	E	E	E 
F	E	E	Es`   V	
VV-V?WW#W5=X	
V
V*-
V<?
W
W #
W25
X
Xc            	          / SQn / SQn[        XSS9u  p#n[        U[        R                  SSS/5        / SQn / S	Qn[        XSS9u  p#n[        U[        R                  SS
SSS/5        g )Nr   r   r   r   r-   r-   r   皙?r   333333?ffffff?r\   Tr   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   r2   inf)rJ   rH   rv   ru   r   s        rK    test_roc_curve_drop_intermediater     sl    F,G$VMCjj2663S*AB 5FOG$VMCjj2663S#s*KLrM   c                     / SQn / SQn[         R                  " SS5      n[        XUS9u  p4n[         R                  " U5      S:  R	                  5       S:X  d   e[         R                  " U5      S:  R	                  5       S:X  d   eg )Nr   r   r-   r-   r-   )r   r   333333?皙?r   r      sample_weightr   )r2   repeatr   diffrR   )rJ   rH   r   ru   rv   rw   s         rK   !test_roc_curve_fpr_tpr_increasingr     sv     F'GIIc1%MF=IKCaGGCL1!!#q(((GGCL1!!#q(((rM   c                  &   SS/n SS/n[        [        X5      S5        SS/n SS/n[        [        X5      S5        / SQn / SQn[        [        X5      S5        SS/n SS/n[        [        X5      S5        / SQn / SQn[        [        X5      S5        g )Nr   r-   r   r-   r   r   r   r   r   r-   )r   r
   )r   rA   s     rK   test_aucr     s    	
AA	
AAc!i-	
AA	
AAc!i-AAc!i-	
AA	
AAc!i+AAc!i-rM   c                     [         R                  " [        5         [        / SQSS/5        S S S 5        [         R                  " [        5         [        S/S/5        S S S 5        / SQn / SQnSR	                  [
        R                  " U 5      5      n[         R                  " [        [        R                  " U5      S9   [        X5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)	Nr   r   r   r   )r&   r-   r      )r            z+x is neither increasing nor decreasing : {}r   )	r   r   r   r
   formatr2   r   reescape)r   rA   error_messages      rK   test_auc_errorsr     s    	z	"Oc3Z( 
# 
z	"SEC5 
# 	AAAHHRSUM	z=)A	BA	 
C	B 
#	" 
#	" 
C	Bs#   CC%?C6
C"%
C36
Dzy_true, labels)r   r-   r   r&   r   r-   r&   )abr   c)r   r   r   c           
      V   [         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        / SQ/ SQ5      n[        / SQ/ SQ5      nX4-   S	-  n[        / S
Q/ SQ5      n[        / SQ/ SQ5      nXg-   S	-  n[        SS/SS/5      n	[        SS/SS/5      n
X-   S	-  nXX-   U-   S-  n[        [        XUSS9U5        XXU/n/ SQn[         R                  " XS9n[        [        XUSSS9U5        Sn[
        R                  " [        US9   [        XUSS S9  S S S 5        g ! , (       d  f       g = f)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   )r2   r   r   r   r   r   r   NotImplementedError)rJ   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                    rK   #test_multiclass_ovo_roc_auc_toydatar    sI    xx	/+<mLH Y(89HY8H +q0 Y7HY(89H +q0 aVc3Z0HaVc3Z0H +q0 -?BRRVWWfv5I $7GHK#JKDV
	
 		 MM	*-	@fv5RVW 
A	@	@s   D
D()r   r&   r   r&   )r   dr   r  )r   r   r  c           
          [         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        / SQ/ SQ5      n[        / SQ/ SQ5      nX4-   S	-  n[        [        XUS
S9U5        [        [        XUS
SS9U5        g )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  r2   r   r   r   )rJ   r   r  r  r  	ovo_scores         rK   *test_multiclass_ovo_roc_auc_toydata_binaryr  '  s~     xx	/+<oNH \+@AH\+@AH$)Ifv5I9
 V
	
 		rM   )r   r-   r&   r&   )r   r   r   r   c           
         [         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        / SQUS S 2S4   5      n[        / SQUS S 2S4   5      n[        / S	QUS S 2S
4   5      n[        [        XSUS S9X4U/5        X4-   U-   S-  n[        [        XSUS9U5        US-  US-  -   US-  -   n[        [        XSUSS9U5        g )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&   ovr)r   r   r   g      @)r   r   r   r   r   r  )rJ   r   r  out_0out_1out_2result_unweightedresult_weighteds           rK   #test_multiclass_ovr_roc_auc_toydatar$  K  s     xx	/?OLH
 ,A7E,A7E,A7EfE&RVW	u .#5fE&I dlUT\1ECK?O%
	
 		rM   zmulti_class, average))r  r   )r  r   )r   r   c           	      .   [         R                  " / SQ5      n/ SQ/ SQ/ SQ/ SQ/n[        [        X#XS9S5        / SQ/ SQ/ SQ/ SQ/n[        X$XS9S:  d   eS[         R                  " S	5      -  n[        X%XS9[
        R                  " S
5      :X  d   eg )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   )r2   r   r   r   r   r   approx)r   r   rJ   	y_perfecty_imperfecty_chances         rK   0test_perfect_imperfect_chance_multiclass_roc_aucr,  v  s     XXl#F 	 	I f[R 		K 	f{T
		
 bggfo%Hk	s	  rM   c                    U n[         R                  R                  / SQSUS9n[        R                  " U Vs/ s H/  n[         R
                  R                  SX1S9R                  5       PM1     sn5      n[        U/ SQS9n[        UR                  5       UR                  5       5      u  pgn[        Xg5      n	[        XBSS	S
9n
U	[        R                  " U
5      :X  d   eg s  snf )N)       @r\   r     )r   r,   r-   )nrD   r,   r   )classesr  r   r'  )r   	dirichletrvsr2   asarraymultinomialrq   r   r   ravelr
   r   r   r(  )global_random_seedseedrI   y_pred_irJ   y_onehotru   rv   rw   roc_auc_by_handroc_auc_autos              rK   test_micro_averaged_ovr_roc_aucr=    s    D
 __  t$ OFZZ #	
" !!A!ELLN"	
F fi8HHNN,flln=KCa#mO UGTLfmmL9999	
s   6C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                     [         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        R                  " [        U S9   [        XX#S9  S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   r   r   )r2   r   r   r   r   r   )msgrJ   r   r   r  s        rK   *test_roc_auc_score_multiclass_labels_errorrB    sF    R xx	/+<mLH 
z	-fvO 
.	-	-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   rt   z'multi_class must be in \('ovo', 'ovr'\)c                     [        S5      nUR                  SS5      n[        U5      nUR                  SSSS9n[        R
                  " [        U S9   [        XT40 UD6  S S S 5        g ! , (       d  f       g = f)N     r   r   r   r   )r$   randr    r   r   r   r   r   )rA  kwargsrE   rH   y_probrJ   s         rK   #test_roc_auc_score_multiclass_errorrI    sc    N S
!Chhr1oGWF[[AB['F	z	-f// 
.	-	-s   A))
A7c                  4   [        S5      n U R                  S5      n[        R                  " SSS9nSn[        R
                  " [        US9   [        X!5        S S S 5        [        R                  " SSS9n[        R
                  " [        US9   [        X!5        S S S 5        [        R                  " SSSS9n[        R
                  " [        US9   [        X!5        S S S 5        g ! , (       d  f       N= f! , (       d  f       Nc= f! , (       d  f       g = f)NrD  
   r5   dtyper   r   rO   )
r$   rF  r2   r   r   r   r   r   r   r   )rE   rI   rJ   warn_messages       rK   test_auc_score_non_binary_classrO  4  s     S
!CXXb\FXXb&F	5  
,L	Af% 
BWWRu%F	,L	Af% 
BWWR5)F	,L	Af% 
B	A 
B	A 
B	A 
B	As$   C'C8D	'
C58
D	
D
curve_funcc                     [        S5      nUR                  SSSS9nUR                  S5      nSn[        R                  " [
        US9   U " X#5        S S S 5        g ! , (       d  f       g = f)NrD  r   r   rK  r   z"multiclass format is not supportedr   )r$   r   rF  r   r   r   )rP  rE   rJ   rI   rA  s        rK   &test_binary_clf_curve_multiclass_errorrR  I  sT    
S
!C[[AB['FXXb\F
.C	z	-6" 
.	-	-s   		A
A)c                    Sn[         R                  " [        US9   U " [        R                  " SS/SS9SS/5        S S S 5        [         R                  " [        US9   U " [        R                  " SS/[
        S9SS/5        S S S 5        / S	QnU " / S
QU5      nU " / SQU5      n[        X45       H$  u  pV[        R                  R                  XV5        M&     g ! , (       d  f       N= f! , (       d  f       Nm= f)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<U1rL  r   r\   )r   r\   r   gzG?)r   r-   r-   r   )r   r\   r\   r   )	r   r   r   r2   r   objectziptestingr   )rP  rA  rI   	int_curvefloat_curveint_curve_partfloat_curve_parts          rK   (test_binary_clf_curve_implicit_pos_labelr[  S  s    
	  
z	-288S#Je4sCjA 
. 
z	-288S#Jf5SzB 
.
 #F<0I16:K,/	,G(


"">D -H 
.	- 
.	-s   !C%C*
C'*
C8z.ignore:Support for labels represented as byteslabels_typerg   r   c                     [        SS/U5      nSn[        R                  " [        US9   U " USS/5        S S S 5        g ! , (       d  f       g = f)N   a   bz8Support for labels represented as bytes is not supportedr   r   r\   )r   r   r   	TypeError)rP  r\  r   rA  s       rK   .test_binary_clf_curve_implicit_bytes_pos_labelra  l  sA      tk:F
DC	y	,6C:& 
-	,	,s	   ?
Ac                     / SQn/ SQn/ SQnU " XUS9nU " US S US S US S S9n[        XE5       H  u  pg[        Xg5        M     g )Nr   )r   r   r   r   r   )r-   r-   r-   r   r   r   rO   )rU  r   )rP  rJ   rH   r   result_1result_2arr_1arr_2s           rK   (test_binary_clf_curve_zero_sample_weightrg  x  sY    F'G%M&GH&"+ws|=QTRTCUVHH/% 0rM   c                    [        SS9u  pn[        XU 5        [        USS  USS  U S9u  pEnUS   S:X  d   eUS   USS  R                  5       :X  d   eSU[        R
                  " US:H  5      '   UR                  5       n[        XU 5        [        Xq5        / SQn/ S	Qn	[        XU S9u  pEnU (       a+  [        U/ S
Q5        [        U/ SQ5        [        U/ SQ5        O*[        U/ SQ5        [        U/ SQ5        [        U/ S	Q5        UR                  UR                  :X  d   eUR                  UR                  S-   :X  d   eg )NTr   r-   r   r   r\   rO   r-   r   r   r-   )r-   r&   r   r   )r   QUU?r\   r\   )r\   r   r   r   )r-   r&   r   )r   rj  r   r\   r\   )r\   r   r   r   r   )
rL   _test_precision_recall_curver   meanr2   wherecopyr   r   r   )
r   rJ   rw   rH   rD   rr   y_true_copyr   predict_probass
             rK   test_precision_recall_curverr    s.   (5Fw $7 %VABZPTUGA!Q43;;Q46!":??$$$$ %'F288FaK !++-K $7{+F!N$VtTGA!67/09%;<45<(66QVV66QVVaZrM   c                    [        XUS9u  p4n[        X5      n[        USS5        [        U[        X5      5        [	        [        X5      USS9  UR                  UR                  :X  d   eUR                  UR                  S-   :X  d   e[        U [        R                  " U5      US9u  p4nUR                  UR                  :X  d   eUR                  UR                  S-   :X  d   eg )Nr   grh|?r   r&   r   r-   )	r   rm   r   r   r   re   r   r2   
zeros_like)rJ   rH   r   rD   ro  r   precision_recall_aucs          rK   rk  rk    s    -fQUVA*26C2E1=5fF 6+-A1 66QVV66Z__q((((-g&$A* 66QVV66Z__q((((rM   c           
         [         R                  " SS9   SS/nSS/n[        XU S9u  p4n[        X5      n[	        U/ SQ5        [	        U/ SQ5        [        US5        SS/nSS/n[        XU S9u  p4n[        X5      n[	        U/ S	Q5        [	        U/ S
Q5        [        US5        SS/nSS/n[        XU S9u  p4n[        X5      n[	        USS/5        [	        USS/5        [        US5        SS/nSS/n[        XU S9u  p4n[        X5      n[	        U/ SQ5        [	        U/ SQ5        [        US5        SS/nSS/n[        XU S9u  p4n[        X5      n[	        USS/5        [	        USS/5        [        US5        SS/nSS/n[        R                  " [        SS9   [        XU S9u  p4nS S S 5        [        R                  " [        SS9   [        X5      nS S S 5        [        U/ SQ5        [        U/ SQ5        [        US5        SS/nSS/n[        XU S9u  p4n[        [        X5      S5        [	        U/ SQ5        [	        U/ SQ5        [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        R                  " [        SS9   [        [        XSS9S5        S S S 5        [        R                  " [        SS9   [        [        XSS9S5        S S S 5        [        [        XSS9S5        [        [        XSS9S5        [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        R                  " [        SS9   [        [        XSS9S5        S S S 5        [        R                  " [        SS9   [        [        XSS9S5        S S S 5        [        [        XSS9S5        [        [        XSS9S5        [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        R                  " [        SS9   [        [        XSS9S5        S S S 5        [        [        XSS9S5        [        R                  " [        SS9   [        [        XSS9S5        S S S 5        [        R                  " [        SS9   [        [        XSS9S5        S S S 5        [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        [        [        XSS9S5        S S S 5        [         R                  " SS9   [         R                  " SS/SS//5      n[         R                  " SS/SS//5      n[        R                  " [        SS9   [        [        XSS9S5        S S S 5        S S S 5        g ! , (       d  f       GN\= f! , (       d  f       GNA= f! , (       d  f       GNw= f! , (       d  f       GNS= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GNg= f! , (       d  f       GNC= f! , (       d  f       GN== f! , (       d  f       N= f! , (       d  f       g = f)Nraise)allr   r-   r   )r   r-   r-   r   r\   )r   r   r\   r  r   r   r   r   z!No positive class found in y_truer   r   )r\   r\   r\   )r-   r   r   r   r   r   r   r   ignore)r2   errstater   r   r   r   r   r   UserWarningr   r   )r   rJ   rH   rD   ro  rw   auc_prcs          rK   #test_precision_recall_curve_toydatar}    s   		!Qa&(DQa)&:!![1!!Y/GS)Qa&(DQa)&:!!_5!!_5 	GS)Qa&(DQa)&:!!c1X.!!c1X.GS)Qa&(DQa)&:!![1!!Y/GS)Q*(DQa)&:!!c1X.!!aX.GS)Q,\\+-PQ,VPTUGA! R\\+-PQ-f>G R9%9%#Q,(DQa3FDcJ!!_5!!]3 Aq6Aq6*+((QFQF+,\\+-PQ'I3 R \\+-PQ'Lc R 	#FYG	
 	/QSVWAq6Aq6*+((QFQF+,\\+-PQ'I3 R \\+-PQ'Lc R 	#FYG	
 	/QSVWAq6Aq6*+((QFQF+,#FWEs	
 	#FZH#	
 	#FYG	
 	#FWEs	
 Aq6Aq6*+((QFQF+,\\+-PQ'I3 R 	#FZH#	
 \\+-PQ'KS R \\+-PQ'I3 R
 Aq6Aq6*+((QFQF+,/QSVW#FZH#	
 	#FYG	
 	/QSVWAq6Aq6*+((S#Jc
34#FWEs	
 	#FZH#	
 	#FYG	
 	#FWEs	
i 
"p 
	"Aq6Aq6*+((QFQF+,\\+-PQ'La R	 
#	"S RQQQ  RQ RQ RQ RQ4 RQ RQ RQs 
"	!x RQ	 
#	"s  F
^$\-!^$\C^$!\&6!^$\8,B^$-]
!^$#]8D	^$].5^$^  !^$^D^$
A_^61_
\	^$
\#	^$&
\5	0^$8
]	^$

]	^$
]+	&^$.
]=	8^$ 
^	
^$
^!	^$$
^36
_	 _
_c                     / SQn / SQn[        XSS9u  p#n[        U/ SQ5        / SQn / SQn[        XSS9u  p#n[        U/ SQ5        / S	Qn / S
Qn[        XSS9u  p#n[        USS/5        / SQn / S
Qn[        XSS9u  p#n[        U/ S
Q5        g)z9Check the behaviour of the `drop_intermediate` parameter.r   r   Tr   )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-   N)r   r   )rJ   rH   ri   rj   r   s        rK   -test_precision_recall_curve_drop_intermediater  _  s    F,G$:4%!Iz J0 5FOG$:4%!Iz J >? F"G$:4%!Iz Jc
+ F"G$:4%!Iz J 45rM   c                      [         R                  " S[        S9n SU S S S2'   [         R                  " S5      n[	        X5      S:X  d   eg )Nr   rL  r-   r   r   )r2   r   r5   r   r   rJ   rH   s     rK   &test_average_precision_constant_valuesr    sC    
 XXc%FF3Q3KggclG #63t;;;rM   c                      [         R                  " SS/5      n [         R                  " SS/5      nSn[        R                  " [        US9   [        XSS9  S S S 5        g ! , (       d  f       g = f)Nr   r-   z>pos_label=2 is not a valid label. It should be one of \[0, 1\]r   r&   rU   r2   r   r   r   r   r   rJ   rI   err_msgs      rK   4test_average_precision_score_binary_pos_label_errorsr    sO    XXq!fFXXq!fFOG	z	1!< 
2	1	1s   A  
A.c                  
   [         R                  " SS/SS/SS/SS//5      n [         R                  " SS/SS/SS/SS//5      nSn[        R                  " [        US9   [        XSS	9  S S S 5        g ! , (       d  f       g = f)
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      rK   8test_average_precision_score_multilabel_pos_label_errorsr    s     XX1v1v1v1v67FXXSzC:SzC:FGF	6  
z	1!< 
2	1	1s    A44
Bc            	         [         R                  " / SQ5      n [         R                  " / SQ/ SQ/ SQ/ SQ/ SQ/ SQ/5      nSn[        R                  " [        US9   [        XSS	9  S S S 5        g ! , (       d  f       g = f)
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      rK   8test_average_precision_score_multiclass_pos_label_errorsr    sf    XX()FXX	
	F	6  
z	1!< 
2	1	1s   A00
A>c                  L   [        SS9u  pn[        X5      n[        U SU-  5      n[        U SU-  5      n[        XS-
  5      nX4:X  d   eX5:X  d   eX6:X  d   e[        X5      n[        U SU-  5      n[        U SU-  5      n	[        XS-
  5      n
Xx:X  d   eXy:X  d   eXz:X  d   eg )NTr   r   gư>rK  )rL   r   r   )rJ   rw   rH   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              rK   test_score_scale_invariancer    s     )5FwF,G%fcGm<'w?#FbL9O''')))%%%$V5F.vsW}E0H,Vr\BN%%%'''###rM   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                 L    [        X5      u  pEn[        XB5        [        XS5        g Nr   r   rJ   rH   expected_fprexpected_fnrru   fnrrw   s          rK   test_det_curve_toydatar    s"    6 F,KCaC&C&rM   )rJ   rH   r  r  r   )r-   r-   r   r   )r-   r-   r   c                 J    [        XUS9u  pVn[        XR5        [        Xc5        g )Nr   r  )rJ   rH   r  r  r   ru   r  rw   s           rK    test_det_curve_drop_intermediater    s%    & F?PQKCaC&C&rM   r   r   r   c                 L    [        X5      u  pEn[        XB5        [        XS5        g r  r  r  s          rK   test_det_curve_tie_handlingr    s"     F,KCaC&C&rM   c                  T    [        [        / SQ/ SQ5      [        / SQ/ SQ5      5        g )Nr   r   r   )r   r   r   r   r-   r-   )r   r    rM   rK   test_det_curve_sanity_checkr  !  s     )[)$&<=rM   rH   )r   r   r   r   r-   c                     [        / SQ[        R                  " SU 5      S9u  pn[        USS/5        [        USS/5        [        X0[        R                  /5        g )N)r   r-   r   r-   r   r-   r   r  r-   r   )r   r2   r   r   r   )rH   ru   r  rk   s       rK   test_det_curve_constant_scoresr  )  sP    #!2771g+>Ci C!Q C!Q I01rM   rJ   )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                 P    [        X S9u  pn[        US/5        [        US/5        g )Nr  r   r  )rJ   ru   r  rw   s       rK   test_det_curve_perfect_scoresr  4  s*     6:KCaC!C!rM   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                     [         R                  " [        US9   [        X5        S S S 5        g ! , (       d  f       g = f)Nr   )r   r   r   r   r  s      rK   test_det_curve_bad_inputr  E  s&      
z	1&! 
2	1	1   /
=c                  P   S/S-  S/S-  -   n [         R                  " / SQ5      nSU-
  n[        U USS9u  p4n[        U USS9u  pgnUS   [        R                  " S	5      :X  d   eUS   [        R                  " S
5      :X  d   e[        X7S S S2   5        [        XFS S S2   5        g )Nr  r   r  r   )
r   r   r   r   r   r   r   r   r   r   r-   r  r   r   r   rO   )r2   r   r   r   r(  r   )	rJ   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	            rK   test_det_curve_pos_labelr  Y  s    Z!^|nq00FHH%WX114=51NM
 AJA=,= v}}S1111Q6==#5555 Ntt$<=Ntt$<=rM   c                    [        U " SS//SS//5      S5        [        U " SS//SS//5      S5        [        U " SS//SS//5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / S	Q// SQ/5      S5        [        U " / S
Q// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / S	Q// SQ/5      S5        [        U " / S
Q// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / S	Q// SQ/5      S5        [        U " / S
Q// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " SS//SS//5      S5        [        U " SS//SS//5      S5        [        U " SS//SS//5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / S	Q// SQ/5      S5        [        U " / S
Q// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        [        U " / SQ// SQ/5      S5        g )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    rK   check_lrap_toyr  s  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RrM   c           	         [        S5      n[        SS5       H  nUR                  SU4S9n[        R                  " U5      n[        R
                  " SU45      nU " XS5      S:X  d   eU " XT5      S:X  d   e[        R                  " SU45      nU " XS5      S:X  d   eU " XT5      S:X  a  M   e   [        U " S/S/S/S//S/S/S/S//5      S5        g )Nr   r&   r   r-   r   r\   r   )r$   r^   uniformr2   rt  r   r   r   )r  r,   n_labelsrH   y_score_tiesrJ   s         rK   !check_zero_or_all_relevant_labelsr    s    %a(L!QK&&QM&:}}W- 1h-(&*c111&/3666 !X'&*c111&/3666   QC!qcA3'3%#u)EFrM   c                    [         R                  " [        5         U " / SQ/ SQ5        S S S 5        [         R                  " [        5         U " / SQ/ SQ/ SQ/ SQ/5        S S S 5        [         R                  " [        5         U " / SQ/ SQ/ SQ/ SQ/5        S S S 5        [         R                  " [        5         U " SS/SS//SS/5        S S S 5        [         R                  " [        5         U " SS/SS//SS//5        S S S 5        [         R                  " [        5         U " SS/SS//S/S//5        S S S 5        [         R                  " [        5         U " SS//SS/SS//5        S S S 5        [         R                  " [        5         U " S/S//SS/SS//5        S S S 5        [         R                  " [        5         U " SS/SS//S/S//5        S S S 5        g ! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GNt= f! , (       d  f       GNR= f! , (       d  f       GN/= f! , (       d  f       GN= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)	Nr   )r   r   r   r   )r   r   r   )r   r   r   )r   r   r   r   r-   )r   r   r   r  s    rK   check_lrap_error_raisedr    s   	z	"9./ 
#	z	"90/?ST 
#	z	"//R	
 
# 
z	"QFQF#aV, 
#	z	"QFQF#q!fX. 
#	z	"QFQF#qcA3Z0 
#	z	"QF8q!fq!f-. 
#	z	"QC!:AA/0 
#	z	"QFQF#qcA3Z0 
#	") 
#	"	"	"	"	" 
#	"	"	"	"	"	"	"	"	"	"	"sk   G G2H9H-H("H:III. 
G/2
H
H
H%(
H7:
I	
I
I+.
I<c           	         [        SS5       Hy  n[        R                  " SU45      n[        SU5       HN  n[        X-
  5       H:  n[        R                  " SU45      nSUSXDU-   24'   [	        U " XR5      X1-  5        M<     MP     M{     g )Nr&   rK  r-   r   )r^   r2   r   r   r   )r  r  rH   
n_relevantrV   rJ   s         rK   check_lrap_only_tiesr    s     !RL''1h-(  8,JX231h-045q#j 0001#Jv$?AVW 4 -	 !rM   c                   ^^ [        SS5       H  nU[        R                  " U5      R                  SU45      S-   -
  n[        R                  " SU45      nSUS'   SUS'   [        U " X25      SU-  S-   S-  5        [        SU5       Hk  m[        UT-
  5       HV  m[        R                  " SU45      nSUSTTT-   24'   [        U " X25      [        UU4S j[        T5       5       5      5        MX     Mm     M     g )Nr&   rK  r-   )r   r   )r   rO   r   c              3   F   >#    U  H  nUS -   TU-   S -   T-  -  v   M     g7f)r-   Nr  ).0ro  r  rV   s     rK   	<genexpr>>check_lrap_without_tie_and_increasing_score.<locals>.<genexpr>  s.      !2A QC!GaK:#=>!2s   !)r^   r2   r3   rQ   r   r   rR   )r  r  rH   rJ   r  rV   s       @@rK   +check_lrap_without_tie_and_increasing_scorer    s     !RLbii1991h-H1LM 1h-(tuJv7!h,:Ja9OP  8,JX
231h-045q#j 0001#v/ !&z!2  4 - !rM   c                   ^^ [        X5        [        U 5      n [        U5      nU R                  u  p#[        R                  " U45      n[        U5       H  n[        R                  " X   SS9u  pgUR                  nX-
  m[        R                  " TUS-   S9R                  5       n	U	T   mX   R                  5       S   n
U
R                  S:X  d  U
R                  U:X  a  SXE'   M  SXE'   U
 H+  m[        UU4S jU
 5       5      nXE==   UTT   -  -  ss'   M-     XE==   U
R                  -  ss'   M     UR                  5       $ )z8Simple implementation of label ranking average precisionT)return_inverser-   )	minlengthr   r   c              3   :   >#    U  H  nTU   TT   :*  v   M     g 7fr  r  )r  ro  labelranks     rK   r  _my_lrap.<locals>.<genexpr>%  s      JAaDK!7s   )r#   r"   r1   r2   emptyr^   rP   r   bincountcumsumnonzerorR   rl  )rJ   rH   rB   r  ra   rb   unique_rankinv_rankn_ranks	corr_rankrelevantn_ranked_abover  r  s               @@rK   _my_lrapr  	  s,   F, F'"G ,,IHHi\"E9 !#		'*T J""! KK!<CCE	9$$&q)==A(!:EHE ! J JJN He44H  	HMM!5 8 ::<rM   r   c                 h   [        SSUUUS9u  pE[        UR                  S   UR                  S   US9n[        US5      (       a  UR	                  5       n[        XV5      n[        XV5      n[        Xx5        [        U5      nUR                  X!4S9n[        XV5      n[        XV5      n[        Xx5        g )Nr-   F)rC   allow_unlabeledr,   	n_classesrB   r   )n_componentsrC   r,   toarrayr   )
r   r   r1   hasattrr  r   r  r   r$   r  )	r  r  rB   r,   rw   rJ   rH   
score_lrapscore_my_lraps	            rK   %check_alternative_lrap_implementationr  /  s     /!IA $\\!_<<?!G w	""//#6vGJV-M
2 &l3L""(>"?G6vGJV-M
2rM   checkfuncc                     U " U5        g r  r  )r  r  s     rK   test_label_ranking_avpr  O  s     
$KrM   c                  "    [        [        5        g r  )r  r   r  rM   rK   test_lrap_error_raisedr  ]  s    ABrM   rB   )r-   r&   r   rE  r  )r&   r   rK  r,   c                 &    [        [        XU5        g r  )r  r   )rB   r  r,   s      rK   $test_alternative_lrap_implementationr  a  s     *-y\rM   c                  h   [         R                  " / SQ/ SQ/ SQ/[        S9n [         R                  " / SQ/ SQ/ SQ/5      n[         R                  " / SQ5      n[         R                  " / S	Q5      n[        [	        XUS
9[         R
                  " X2-  5      [         R
                  " U5      -  5        g )Nr  ri  r  rL  )r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r\   )r\   r\   r   r   )r2   r   boolr   r   rR   )rJ   rH   samplewise_lrapsr   s       rK   &test_lrap_sample_weighting_zero_labelsr  j  s     XX|\<@MFhh	35IJG xx 01HH_-M-=	
 	}/0266-3HH	rM   c            	         [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q// SQ/5      S5        [        [        / S
Q// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q// SQ/5      S5        [        [        / S
Q// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q// SQ/5      S5        [        [        / S
Q// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q/ SQ// SQ/ SQ/5      S5        [        [        / S	Q/ SQ/ S
Q// SQ/ SQ/ SQ/5      S5        [        [        / S	Q/ SQ/ S
Q// SQ/ SQ/ SQ/5      S5        g )Nr   r-   r   r   r&   r  r  r   r   r   r   r   r   r   r  r  r  r   g      $@r   r-   r   r.  r   rK  r  r>  g@r   r-   r   r   r   r  rM   rK   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	
 		rM   c                     [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q// SQ/5      S5        [        [        / S
Q// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        g )Nr   r   r-   r&   r  r  r   r   r   r   r   r   r   r  r  r  rM   rK   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KrM   zy_true, y_scorec                     [         R                  " [        SS9   [        X5        S S S 5        g ! , (       d  f       g = f)Nz'Expected 2D array, got 1D array insteadr   )r   r   r   r   r  s     rK   test_coverage_1d_error_messager    s(     
z)S	Tv' 
U	T	Tr  c            	      h   [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q// SQ/5      S5        [        [        / S
Q// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ/ SQ// SQ/ SQ/5      S5        [        [        / SQ/ SQ/ S	Q// SQ/ SQ/ SQ/5      S5        [        [        / SQ/ SQ/ S	Q// SQ/ SQ/ SQ/5      S5        g )Nr   r-   r   r   r   r  r   r   r   r   r\   r   r   r  r  r  r  r  r   r  r>  r  r   r   r  rM   rK   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	
 	"	rM   csr_containerc                 |    [        [        U " [        R                  " / SQ/ SQ/5      5      / SQ/ SQ/5      S5        g )Nr   r   r  r  r   )r   r   r2   r   )r  s    rK   test_label_ranking_loss_sparser     s4    "((Iy#9:;mY=W	
 		rM   c                  x   [         R                  " [        5         [        SS/SS//SS/5        S S S 5        [         R                  " [        5         [        SS/SS//SS//5        S S S 5        [         R                  " [        5         [        SS/SS//S/S//5        S S S 5        [         R                  " [        5         [        SS//SS/SS//5        S S S 5        [         R                  " [        5         [        S/S//SS/SS//5        S S S 5        [         R                  " [        5         [        SS/SS//S/S//5        S S S 5        g ! , (       d  f       GN+= f! , (       d  f       GN= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)Nr   r-   )r   r   r   r   r  rM   rK   $test_ranking_appropriate_input_shaper  
  sI   	z	"QFQF+aV4 
#	z	"QFQF+q!fX6 
#	z	"QFQF+qcA3Z8 
#	z	"QF8q!fq!f%56 
#	z	"QC!:AA'78 
#	z	"QFQF+qcA3Z8 
#	" 
#	"	"	"	"	"	"	"	"	"	"	"sG   EE&
E8F	;F4F+
E#&
E58
F	
F
F(+
F9c                     [        [        SS//SS//5      S5        [        [        SS//SS//5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / SQ// SQ/5      S5        [        [        / S	Q// SQ/5      S5        [        [        / S
Q// SQ/5      S5        g )Nr-   r   r   r   r  r   r   r   r   r   r
  r  rM   rK   test_ranking_loss_ties_handlingr    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OrM   c                      [        SSS9u  pU* S-   n[        X5        [        R                  R	                  S5      R                  S5      u  p[        X5        g Nr   rK  )r,   r  r-   )r&   r   rK  )r   _test_dcg_score_forr2   r6   r7   random_samplerw   rJ   rH   s      rK   test_dcg_scorer  &  sL    .ALIAgkG(ii++A.<<\JOF(rM   c                 .   [         R                  " [         R                  " U R                  S   5      S-   5      n[	        X 5      n[	        X5      nXC:*  R                  5       (       d   e[	        X SS9U:*  R                  5       (       d   eUR                  U R                  S   4:X  d   eUR                  U R                  S   4:X  d   eU[        R                  " [         R                  " U 5      S S 2S S S24   U-  R                  SS95      :X  d   eg )Nr-   r&   r   kr   rO   axis)
r2   log2r3   r1   r   rx  r   r(  sortrR   )rJ   rH   discountidealra   s        rK   r  r  .  s    wwryya1A56Hv.Ev/EN!!!!v3u<AACCCC;;6<<?,,,,;;6<<?,,,,FMM2776?1dd7#;h#F"K"KQR"K"STTTTrM   c            	         [         R                  " [         R                  " S5      /5      n [         R                  " U R                  5      n[        X5      n[        XSS9nS[         R                  " [         R                  " SS5      5      -  nU[        R                  " UR                  5       U R                  5       -  /5      :X  d   eU[        R                  " X@S S 2S S S24   -  R                  5       /5      :X  d   eSUSS	S 24'   [        X5      n[        XSS9nU[        R                  " X@S S 2S S S24   -  R                  5       /5      :X  d   eU[        R                  " US S R                  5       U SS	S 24   R                  5       -  USS  R                  5       U SS S	24   R                  5       -  -   /5      :X  d   eg )
Nr   Tignore_tiesr-   r&   r   rO   r   r   )r2   r4  r3   r   r1   r   r  r   r(  rR   rl  )rJ   rH   dcgdcg_ignore_ties	discountss        rK   test_dcg_tiesr(  9  s   ZZ1'Fhhv||$G
V
-C(dKOBGGBIIaO,,I&--6;;=!@ ABBBBfmmiDbD/.I-N-N-P,QRRRRGAqrEN
V
-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	
   rM   c            
          [         R                  " S5      R                  S5      n [        X SSS9[        R
                  " [        X SSS95      :X  d   eg )N   )r&   r   r   T)r  r$  )r2   r3   rQ   r   r   r(  )r   s    rK   test_ndcg_ignore_ties_with_kr+  M  sL    
		"f%AaaT2fmm11$/7   rM   c                      [         R                  " / SQ/5      n [         R                  " / SQ/5      nSn[        R                  " [        US9   [        X5        SSS5        g! , (       d  f       g= f)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)r2   r   r   r   r   r   )rJ   rH   r   s      rK    test_ndcg_negative_ndarray_errorr-  T  sO    XX89:Fhh678GP	z)9	:6# 
;	:	:s   A##
A1c                     [         R                  " S5      R                  SS5      n U [         R                  R	                  S5      R                  SSU R                  S9-   n[        X5      n[        XSS	9nU[        R                  " U5      :X  d   eU[        R                  " S
5      :X  d   eUS-  n[        X5      [        R                  " S
5      :X  d   eg )NF   r   rK  r   皙ɿr   r   Tr#  r\   r/  )
r2   r3   rQ   r6   r7   r  r1   r   r   r(  )rJ   rH   ndcgndcg_no_tiess       rK   test_ndcg_invariantr3  ]  s    YYr]""1b)Fryy,,Q/77c7UUGf&Df4@L6==....6==%%%%tOGf&&--*<<<<rM   r$  c           
      &   S[         R                  " S5      S S -  n[         R                  " [         R                  " SSS5      S5      nU[         R                  R                  S5      R                  SS	UR                  S
9-   n[        XU S9[        R                  " S[         R                  " [         R                  " SS5      5      -  5      :X  d   e[        XU S9[        R                  " S[         R                  " [         R                  " SS5      5      -  5      :X  d   e[        XU S9[        R                  " S[         R                  " [         R                  " SS5      5      -  5      :X  d   e[        XSU S9[        R                  " S[         R                  " [         R                  " SS5      5      -  5      :X  d   e[        XU S9[        R                  " S[         R                  " [         R                  " SS5      5      -  R                  5       5      :X  d   e[!        XU S9[        R                  " S[         R                  " [         R                  " SS5      5      -  R                  5       5      :X  d   eS[         R"                  " S5      -  nS[         R                  " [         R                  " SS5      5      -  R%                  5       n[        XU S9[        R                  " U[         R"                  " S5      -  5      :X  d   e[        XU S9[        R                  " [         R"                  " S5      5      :X  d   e[!        XU S9[        R                  " U5      :X  d   e[        XU S9[        R                  " S5      :X  d   eg )Nr   r   r   r   rO   )r   r-   r   r0  r   r   r#  r&   r-   rK  )log_baser$  )r   r   	   r\   )r2   eyetiler3   r6   r7   r  r1   r   r   r(  r  r   log10r   rl  r   r   rR   )r$  rJ   rH   y_score_noisyexpected_dcg_scores        rK   test_ndcg_toy_examplesr<  h  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 f;?6==	
RWWRYYq!_%	%++-D    V+>&--	
RWWRYYq!_%	%++-C     Fbggbii1o66;;=[	)BGGAJ6	78 8 8 [	rwwqz	"# # # V+>&--C    f;?6==QTCUUUUrM   c                      Sn [         R                  " [        U S9   [        S//S//5        SSS5        g! , (       d  f       g= f)zdCheck 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    rK   test_ndcg_error_single_documentr>    s:    	  
z	1QC5A3%  
2	1	1s	   6
Ac                      [        SSS9u  pU* S-   n[        X5        [        R                  R	                  S5      R                  S5      u  p[        X5        g r  )r   _test_ndcg_score_forr2   r6   r7   r  r  s      rK   test_ndcg_scorerA    sL    .ALIAgkG)ii++A.<<\JOF)rM   c                    [        X 5      n[        X5      nX2:*  R                  5       (       d   eU S:H  R                  SS9nX$)    [        R                  " [        R
                  " U) R                  5       5      5      :X  d   eX$   [        R                  " [        R                  " UR                  5       5      5      :X  d   eX4)    [        R                  " [        X5      U)    [        X 5      U)    -  5      :X  d   eX4   [        R                  " [        R                  " UR                  5       5      5      :X  d   eUR                  U R                  S   4:X  d   eUR                  U R                  S   4:X  d   eg )Nr   r-   r  )
r   rx  r   r(  r2   r   rR   r   r   r1   )rJ   rH   r!  ra   all_zeros        rK   r@  r@    sC   /E0EN!!!!!  a (Hv}}RWWxi__5F-GHHHH?fmmBHHX\\^,DEEEEv}}6+XI6
V
,hY
7	8     ?fmmBHHX\\^,DEEEE;;6<<?,,,,;;6<<?,,,,rM   c            	      4   [         R                  " / SQ5      n [        X SS9S:X  d   e[        X SS9S:X  d   e[        R                  " [
        5         [        X SS9(       d   e S S S 5        [        R                  " [
        5         [        X SS9(       d   e S S S 5        [        R                  " [
        5         [        X SS9(       d   e S S S 5        [         R                  " / SQ5      n[        XSS9n[        X5      nX#:X  d   e[        XS	S9S
:X  d   e[        SS9u  pn[         R                  " SSS5       H!  n[        [        XUS9[        XU5      5        M#     g ! , (       d  f       GN= f! , (       d  f       N= f! , (       d  f       N= f)Nr  r-   )rt   gMbP?gg?r   )r   r   r   g{Gz?r   r   Tr   g-C6?r   )
r2   r   r   r   r   r   rL   linspacer   r   )rJ   r  roc_auc_with_max_fpr_oneunconstrained_roc_aucrI   rw   rt   s          rK   test_partial_roc_auc_scorerH    s@   XXl#F3q88871<<<	z	"VT::: 
#	z	"VS999 
#	z	"VQ777 
# xx+,H,VqI)&;#<<<373>>>'t4FA;;tQ*&':"67;	
 + 
#	"	"	"	"	"s$   E&E8=F	&
E58
F	
Fzy_true, k, true_scorer   r&   r   r   c                     [         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        XUS9nU[        R                  " U5      :X  d   eg )Nr  r   r   r   r   r   r   r   r   r   r   r   r   r  r2   r   r   r   r(  )rJ   r  
true_scorerH   ra   s        rK   test_top_k_accuracy_scorerO    sH     hh    		
G !A6EFMM*----rM   zy_score, k, true_score)rO   rO   r-   r-   )rO   r-   rO   r-   )r   r   r   r   )r   r   r   r   c                 *   / SQnU R                  5       S:  a  U R                  5       S::  a  SOSnUS:X  a!  X:  R                  [        R                  5      OUn[        X0US9n[        X55      nXgs=:X  a  [        R                  " U5      :X  d   e   eg )Nr  r   r-   r   r  )	minmaxastyper2   int64r   r	   r   r(  )rH   r  rN  rJ   rk   rI   ra   	score_accs           rK    test_top_k_accuracy_score_binaryrV    s     F{{})gkkmq.@aI78Avg!))"((36F A6Ev.I:z!::::::rM   zy_true, true_score, labels)r   r-   r-   r&   r  )r   r?  r?  r   )r   r   r  r?  labels_as_ndarrayc                     U(       a  [         R                  " U5      n[         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        XSUS9nU[        R
                  " U5      :X  d   eg)z,Test when labels and y_score are multiclass.r  rJ  rK  rL  r&   r  r   N)r2   r4  r   r   r   r(  )rJ   rN  r   rW  rH   ra   s         rK   0test_top_k_accuracy_score_multiclass_with_labelsrZ    s[     F#hh    		
G !AfEEFMM*----rM   c            
         [         R                  " SSSSS9u  p[        XSS9u  p#pE[        SS9nUR	                  X$5        [        X#4XE45       Hl  u  p[        SS5       Vs/ s H  n[        XR                  U 5      US9PM     nn[        R                  " [        R                  " U5      S:  5      (       a  Ml   e   g s  snf )NrK  r/  r   )r  rB   n_informativer,   )r,   r&   r  )r   make_classificationr   r   r;   rU  r^   r   r<   r2   rx  r   )	r@   rA   X_trainX_testy_trainy_testrG   r  scoress	            rK   $test_top_k_accuracy_score_increasingrc    s    ''BQDA (81'M$GW
!
,CGGGW%'89HMaQS
HT1 $5$5a$8A> 	 
 vvbggfo)**** :
s   "#C c                     [         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        XUS9[        R                  " U5      :X  d   eg )N)r   r   r   r   )r-   r   r   r   )r   r   r   r   r  r  rM  )rJ   r  rN  rH   s       rK   test_top_k_accuracy_score_tiesre  (  sC     hh		
G  15z9RRRRrM   z	y_true, kr   c                     [         R                  " / SQ/ SQ/ SQ/ SQ/5      nSn[        R                  " [        US9   [        XUS9nS S S 5        WS:X  d   eg ! , (       d  f       N= f)	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-   )r2   r   r   r   r   r   )rJ   r  rH   r   ra   s        rK   !test_top_k_accuracy_score_warningrg  =  sg     hh    		
G	6  
,4D	E$V: 
FA:: 
F	Es   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                     [         R                  " [        US9   [        XSUS9  S S S 5        g ! , (       d  f       g = f)Nr   r&   rY  )r   r   r   r   )rJ   rH   r   rA  s       rK   test_top_k_accuracy_score_errorri  V  s+    f 
z	-V&A 
.	-	-r  c                     U " / SQ/ SQ/5      n[         R                  " / SQ/ SQ/5      n[        X5      nU[        R                  " S5      :X  d   eg )Nr   r   )r   r   r   r  )r2   r   r   r   r(  )r  rJ   rH   results       rK   Otest_label_ranking_avg_precision_score_should_allow_csr_matrix_for_y_true_inputrl    sG     Iy12Fhh34G26CFV]]5))))rM   metricr1  ))FT)r   r-   )r   r\   )zeroonec                    [         R                  R                  S5      nSUS   pCUR                  XSS9nUR	                  U5      nU " XVUS9n[        U[        5      (       a  [         R                  " U5      (       a   egUu  pn
[         R                  " U5      R                  5       (       a   e[         R                  " U	5      R                  5       (       a   e[         R                  " U
5      R                  5       (       a   eg)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.
*   rK  rO   T)r   replacer  N)	r2   r6   r7   choicerF  
isinstancerS   r   any)rm  r1  rE   rB   rU   rJ   y_probark  metric_1metric_2r   s              rK   #test_ranking_metric_pos_label_typesry    s     ))


#Cwr{yZZZ>Fhhy!GFy9F&%  88F#####)/&J88H%))++++88H%))++++88J'++-----rM   c                     [         R                  R                  U 5      nUR                  SSSS9nUR	                  S5      n[        X#5      u    pE[         R                  " US   5      (       d   eg)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&   rK  r   N)r2   r6   r7   r   rF  r   isinf)r7  rE   rJ   rH   rw   r   s         rK   (test_roc_curve_with_probablity_estimatesr|    sb     ))

 2
3C[[AB['FhhrlG 1Aq88JqM""""rM   )NF)r   rE  r   )r   r   numpyr2   r   scipyr   sklearnr   r   sklearn.datasetsr   sklearn.exceptionsr   sklearn.linear_modelr   sklearn.metricsr	   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_FUNCSrL   rZ   re   rm   r   markparametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r$  r,  r=  rB  rI  rO  rR  r[  filterwarningsra  rg  rr  rk  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-  r3  r<  r>  rA  r@  rH  rO  rV  rZ  rc  re  rg  ri  rl  ry  r|  r  rM   rK   <module>r     s~    	    ! ; 5 3    M 4 0 ;  * .  )#`2<&H8 $/
) 0
)
")&#))8)6oNdM	).&" 	,	+	,	&	/	t$	/X/Xd 	,	+	&	'9: 	,	&	t$	y!	/	D D:( 	,bhh|.DiP/HH)*	
 HH\"	
 1HH)*	
 HH\"F	
 HH)*#J	
 HH\"	
 HH)* 	
 AHH)*	
 AHH)*	
 AHH\"	
}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..(#rM   