
    KiJ                       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  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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:S r;\Rx                  R{                  SSS/5      S 5       r>S r?S r@S rAS rBS rCS rDS  rES! rFS" rGS# rHS$ rI\Rx                  R{                  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       rK\Rx                  R{                  S%\R                  " / S+Q5      / S'Q4\R                  " / S,Q5      / S-Q4/5      S. 5       rL\Rx                  R{                  S%\R                  " / S/Q5      S4/ S0QS4/ S/Q/ S'Q4/ S0Q/ S)Q4/5      S1 5       rM\Rx                  R{                  S2/ S3Q5      S4 5       rNS5 rO\Rx                  R{                  S6S7\R                  " / S/Q5      / S8Q4S7\R                  " / S0Q5      / S9Q4S:\R                  " / S+Q5      S4S;\R                  " / S0Q5      / S<Q4S=\R                  " / S/Q5      S S>/4S=\R                  " / S0Q5      S?S@/4SA\R                  " / S/Q5      / SBQ4SA\R                  " / S0Q5      / SCQ4SD\R                  " / SEQ5      / S)Q4SD\R                  " / SCQ5      / S)Q4SD\R                  " / SBQ5      / S'Q4/5      \Rx                  R{                  SFSGSH/5      SI 5       5       rP\Rx                  R{                  SJSKSLSGSM.4SNSLSHSM.4SOSG/ SP.4SQSGSRSS.4ST0 4/5      SU 5       rQSV rR\Rx                  R{                  SW\55      SX 5       rS\Rx                  R{                  SW\55      SY 5       rT\Rx                  R                  SZ5      \Rx                  R{                  SW\\/5      \Rx                  R{                  S[S\S]/5      S^ 5       5       5       rV\Rx                  R{                  SW\55      S_ 5       rW\Rx                  R{                  SSS/5      S` 5       rXSa rY\Rx                  R{                  SSS/5      Sb 5       rZSc r[Sd r\Se r]Sf r^Sg r_Sh r`\Rx                  R{                  Si/ SjQ/ SkQS /S /4/ SjQ/ SlQS /S /4/ SjQ/ SmQS /S /4/ SjQ/ SnQS /S /4/ SoQ/ SkQ/ SpQ/ SqQ4/ SoQ/ SlQ/ SpQ/ SqQ4/ SoQ/ SmQ/ SpQ/ SqQ4/ SoQ/ SnQ/ SpQ/ SqQ4/ SqQ/ SkQSr/S /4/ SqQ/ SlQS /S /4/ SqQ/ SmQS /S /4/ SqQ/ SnQS /S /4/ SsQ/ SkQ/ StQ/ SuQ4/ SsQ/ SlQ/ StQ/ SuQ4/ SsQ/ SmQ/ StQ/ SuQ4/ SsQ/ SnQ/ StQ/ SuQ4/ SvQ/ SkQ/ SwQ/ SxQ4/ SvQ/ SlQ/ SwQ/ SxQ4/ SvQ/ SmQ/ SwQ/ SxQ4/ SvQ/ SnQ/ SwQ/ SxQ4/5      Sy 5       ra\Rx                  R{                  / SzQ/ SsQ/ SkQ/ S{Q/ SuQS4/ SsQ/ SkQ/ S|Q/ SqQS4/ SsQ/ SlQ/ S{Q/ SuQS4/ SsQ/ SlQ/ S|Q/ SqQS4/ SvQ/ SkQ/ SwQ/ SxQS4/ SvQ/ SkQ/ SwQ/ SxQS4/ SvQ/ SlQ/ SwQ/ SxQS4/ SvQ/ SlQ/ SwQ/ SxQS4/5      S} 5       rb\Rx                  R{                  SiS>S /SRSR/S>S /S S>/4S S>/SRSR/S>S /S S>/4/ SjQ/ S~QSRS /S S>/4/ SoQ/ S~QSRS /S S>/4/ SqQ/ S~QS /S /4/ SsQ/ S~Q/ SwQ/ SqQ4/ SvQ/ S~Q/ SwQ/ SkQ4/ SwQ/ S~Q/ SwQ/ SkQ4/5      S 5       rcS rd\Rx                  R{                  S/ SQ5      S 5       re\Rx                  R{                  S/ SQ/ SQ/ SQ/ SQ/ SQ/5      S 5       rf\Rx                  R{                  SS S>// SkQS4/ SqQS SR/S4/ SQ/ SkQS4/ SQ/ SkQS4/ SQ/ SQS4/5      S 5       rgS rhS riS rjS rkS rlS rmS rn SS jro\Rx                  R{                  S\i\m\l\j45      \Rx                  R{                  S\\n45      S 5       5       rpS rq\Rx                  R{                  SS5      \Rx                  R{                  SS5      \Rx                  R{                  S\r" S>5      5      S 5       5       5       rsS rtS ruS rv\Rx                  R{                  S/ SvQ/ S~Q4/ SvQ/ S~Q/4/ SvQ// S~Q4/5      S 5       rwS rx\Rx                  R{                  S\05      S 5       ryS rzS r{S r|S r}S r~S rS rS r\Rx                  R{                  SSS/5      S 5       rS rS rS rS r\Rx                  R{                  S/ SBQS>S4/ SBQSSR4/ SBQSS4/5      S 5       r\Rx                  R{                  S\R                  " / SQ5      S>S>4\R                  " / SQ5      S>SR4\R                  " / SQ5      SS>4\R                  " / SQ5      S>S>4\R                  " / SQ5      S>SR4\R                  " / SQ5      SS>4/5      S 5       r\Rx                  R{                  S\R                  " / SQ5      S/ SBQ4\R                  " / SuQ5      SR/ SBQ4\R                  " / SQ5      SR/ SBQ4\R                  " / SQ5      S/ SQ4/5      \Rx                  R{                  SSS/5      S 5       5       rS r\Rx                  R{                  S/ SBQS>S4/ SBQSSR4/ SBQSS>4/5      S 5       r\Rx                  R{                  S/ SBQS4/ SBQS4/5      S 5       r\Rx                  R{                  S/ SQ/ SQ/ SQ/ SQ/ SQ/SS4/ SBQ/ SQ/ SQ/ SQ/ SQ/SS4/ SQ/ SQ/ SQ/ SQ/ SQ// S0QS4/ SQ/ SQ/ SQ/ SQ/ SQ// S<QS4/ SQ/ SQ/ SQ/ SQ/ SQ// SBQS4/ SQ/ SQ/ SQ/ SQ/ SQ// SQS4S S>// SQ/ SQ/SS4/5      S 5       r\Rx                  R{                  S\05      S 5       r\Rx                  R{                  S\\\\/5      \Rx                  R{                  S/ SQ5      S 5       5       rS rg)    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                 z   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[        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 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                d/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/sklearn/metrics/tests/test_ranking.pymake_predictionrI   :   s-    $$&AAQxq51GGI
		)A
R
 CKKN4qy1}D ))


"C
a9cJ&6778A !
,C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   )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                 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+   NrL   r         ?)r0   rM   rO   argsortrangerQ   )	rG   rE   rR   n_posorderscoreiprecjs	            rH   _average_precisionrb   y   s     		&!!$IFF6&'EJJw"%EnG]FE3w< 9	! D1a!e_9	)CKD % GODME ! =rJ   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[   rQ   )rG   rE   	precisionrecall	thresholdaverage_precisionr_   s          rH   _average_precision_slowrj      sq     $:&#J IyXi()I(6"#F1c)n%q\VYA-FGG &rJ   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   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_roc,_partial_roc_auc_score.<locals>._partial_roc   s    2!Wn%))G-Wn%))CM*1K.K.))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      sc    

" $Fw?Gg'K DDdk"dk2H{H!{-(2EFFGGrJ   c                    [         R                  R                  U 5      nSnUR                  SSSS9nUR	                  SS9n[         R
                  " U5      nX%-
  n[        X45      u  pxpn[        U5      [        U5      s=:X  a0  [        U	5      s=:X  a   [        U
5      s=:X  a  [        U5      :X  d   e   e[        X-   U5        [        Xx-   U5        [        Xx-   U	-   U
-   U5        g)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%CF%T"Cc
s8s3xJ3s8Js3xJ3z?JJJJJCIu%CIu%CIOc)95rJ   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)rI   rW   r   r	   r   r   r   r/   )	r   rG   rt   rE   expected_aucrr   rs   r   roc_aucs	            rH   test_roc_curver      s~     )5Fw(L$VMCj#mGgQ?v!?@99		!!!99
(((((rJ   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+      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fEMCcq6Q;;r7a<<99		!!!99		!!!rJ   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   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Fw$V5Cj KVVW\V+,FF6N38a<(  c:99		!!!99
(((((rJ   c                      [        SS9u  pn[        R                  " [        5         [	        X5        S S S 5        g ! , (       d  f       g = f)NFr   )rI   pytestraises
ValueErrorr   )rG   rt   rE   s      rH   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   )rI   r   r	   r   r/   )rG   rt   rE   rr   rs   r   r   s          rH   test_roc_curve_confidencer     sg    (5Fw$Vs];Cj#mGgtQ799		!!!99
(((((rJ   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(\?)rI   r0   onesr/   r   r	   r   zeros)rG   predrE   trivial_predrr   rs   r   r   s           rH   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
(((((rJ   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   r0   fullrQ   nanr/   )rG   rF   expected_messagerr   rs   r   xs          rH   test_roc_curve_one_labelr   6  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+   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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?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     sl    F,G$VMCjj2663S*AB 5FOG$VMCjj2663S#s*KLrJ   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   )r0   repeatr   diffrO   )rG   rE   r   rr   rs   rt   s         rH   !test_roc_curve_fpr_tpr_increasingr     sv     F'GIIc1%MF=IKCaGGCL1!!#q(((GGCL1!!#q(((rJ   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   r>   s     rH   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-rJ   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	   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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   )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 +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  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$)Ifv5I9
 V
	
 		rJ   )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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fE&RVW	u .#5fE&I dlUT\1ECK?O%
	
 		rJ   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   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_aucr2    s     XXl#F 	 	I f[R 		K 	f{T
		
 bggfo%Hk	s	  rJ   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)       @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_aucrB    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  )r0   r   r   r   r   r   )msgrG   r  r  r
  s        rH   *test_roc_auc_score_multiclass_labels_errorrG    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  rq   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   )rF  kwargsrB   rE   y_probrG   s         rH   #test_roc_auc_score_multiclass_errorrN    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)NrI  
   r3   dtyper   r   rL   )
r$   rK  r0   r   r   r   r   r   r   r   )rB   rF   rG   warn_messages       rH   test_auc_score_non_binary_classrT  I  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)NrI  r   r   rP  r   z"multiclass format is not supportedr   )r$   r   rK  r   r   r   )rU  rB   rG   rF   rF  s        rH   4test_confusion_matrix_at_thresholds_multiclass_errorrW  ^  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<U1rQ  r   rY   )r   rY   r   gzG?)r   r+   r+   r   )r   rY   rY   r   )	r   r   r   r0   r   objectziptestingr   )rU  rF  rF   	int_curvefloat_curveint_curve_partfloat_curve_parts          rH   6test_confusion_matrix_at_thresholds_implicit_pos_labelr`  h  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_typerd   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   rY   )r   r   r   	TypeError)rU  ra  r  rF  s       rH   <test_confusion_matrix_at_thresholds_implicit_bytes_pos_labelrf    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   rL   )rZ  r   )rU  rG   rE   r   result_1result_2arr_1arr_2s           rH   6test_confusion_matrix_at_thresholds_zero_sample_weightrl    sY    F'G%M&GH&"+ws|=QTRTCUVHH/% 0rJ   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   rY   rL   r+   r   r   r+   )r+   r&   r   r   )r}   QUU?rY   rY   )rY   r}   r}   r   )r+   r&   r   )r}   ro  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_curverw    s.   (5Fw $7 %VABZPTUGA!Q43;;Q46!":??$$$$ %'F288FaK !++-K $7{+F!N$VtTGA!67/09%;<45<(66QVV66QVVaZrJ   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   g+?r   r&   r   r+   )	r   rj   r   r
   r   rb   r   r0   
zeros_like)rG   rE   r   rA   rt  r   precision_recall_aucs          rH   rp  rp    s    -fQUVA*26C2E1=5fF 6+-A1 66QVV66Z__q((((-g&$A* 66QVV66Z__q((((rJ   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  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   rt  rt   auc_prcs          rH   #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   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%!Iz J0 5FOG$:4%!Iz J >? F"G$:4%!Iz Jc
+ F"G$:4%!Iz J 45rJ   c                      [         R                  " S[        S9n SU S S S2'   [         R                  " S5      n[	        X5      S:X  d   eg )Nr   rQ  r+   r   r   )r0   r   r3   r   r
   rG   rE   s     rH   &test_average_precision_constant_valuesr    sC    
 XXc%FF3Q3KggclG #63t;;;rJ   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&   rR   r0   r   r   r   r   r
   rG   rF   err_msgs      rH   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      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!< 
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      rH   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ư>rP  )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FwF,G%fcGm<'w?#FbL9O''')))%%%$V5F.vsW}E0H,Vr\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                 L    [        X5      u  pEn[        XB5        [        XS5        g Nr   r   rG   rE   expected_fprexpected_fnrrr   fnrrt   s          rH   test_det_curve_toydatar    s"    6 F,KCaC&C&rJ   )rG   rE   r  r  r   )r+   r+   r}   r   )r+   r+   r   c                 J    [        XUS9u  pVn[        XR5        [        Xc5        g )Nr   r  )rG   rE   r  r  r   rr   r  rt   s           rH    test_det_curve_drop_intermediater    s%    & F?PQKCaC&C&rJ   r   r}   r}   c                 L    [        X5      u  pEn[        XB5        [        XS5        g r  r  r  s          rH   test_det_curve_tie_handlingr  $  s"     F,KCaC&C&rJ   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    rJ   rH   test_det_curve_sanity_checkr  8  s     )[)$&<=rJ   rE   )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   r0   r   r   r   )rE   rr   r  rh   s       rH   test_det_curve_constant_scoresr  @  sP    #!2771g+>Ci 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                 P    [        X S9u  pn[        US/5        [        US/5        g )Nr  r   r  )rG   rr   r  rt   s       rH   test_det_curve_perfect_scoresr  K  s*     6:KCa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                     [         R                  " [        US9   [        X5        S S S 5        g ! , (       d  f       g = f)Nr   )r   r   r   r   r  s      rH   test_det_curve_bad_inputr  \  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   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1NM
 AJA=,= v}}S1111Q6==#5555 Ntt$<=Ntt$<=rJ   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    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           	         [        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   rY   r}   )r$   r[   r   r0   ry  r   r   r   )r  r*   n_labelsrE   y_score_tiesrG   s         rH   !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rJ   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    rH   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&   rP  r+   r   )r[   r0   r   r   r   )r  r  rE   
n_relevantrS   rG   s         rH   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 -	 !rJ   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&   rP  r+   )r   r   )r   rL   r   c              3   F   >#    U  H  nUS -   TU-   S -   T-  -  v   M     g7f)r+   Nr  ).0rt  r  rS   s     rH   	<genexpr>>check_lrap_without_tie_and_increasing_score.<locals>.<genexpr>  s.      !2A QC!GaK:#=>!2s   !)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Jv7!h,:Ja9OP  8,JX
231h-045q#j 0001#v/ !&z!2  4 - !rJ   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  rt  labelranks     rH   r  _my_lrap.<locals>.<genexpr><  s      JAaDK!7s   )r#   r"   r/   r0   emptyr[   rM   r   bincountcumsumnonzerorO   rq  )rG   rE   r?   r  r^   r_   unique_rankinv_rankn_ranks	corr_rankrelevantn_ranked_abover  r  s               @@rH   _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 ::<rJ   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)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     /!IA $\\!_<<?!G w	""//#6vGJV-M
2 &l3L""(>"?G6vGJV-M
2rJ   checkfuncc                     U " U5        g r  r  )r  r  s     rH   test_label_ranking_avpr  f  s     
$KrJ   c                  "    [        [        5        g r  )r  r   r  rJ   rH   test_lrap_error_raisedr  t  s    ABrJ   r?   )r+   r&   r   rJ  r  )r&   r   rP  r*   c                 &    [        [        XU5        g r  )r  r   )r?   r  r*   s      rH   $test_alternative_lrap_implementationr  x  s     *-y\rJ   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"  rn  r  rQ  )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-=	
 	}/0266-3HH	rJ   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   r4  r   rP  r  rC  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                     [        [        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  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                     [         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     rH   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   rY   r   r  r  r  r  r  r  r  r  rC  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                 |    [        [        U " [        R                  " / SQ/ SQ/5      5      / SQ/ SQ/5      S5        g )Nr  r  r  r  r}   )r   r   r0   r   )r  s    rH   test_label_ranking_loss_sparser    s4    "((Iy#9:;mY=W	
 		rJ   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  rJ   rH   $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  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                      [        SSS9u  pU* S-   n[        X5        [        R                  R	                  S5      R                  S5      u  p[        X5        g Nr   rP  )r*   r  r+   )r&   r   rP  )r   _test_dcg_score_forr0   r4   r5   random_samplert   rG   rE   s      rH   test_dcg_scorer  =  sL    .ALIAgkG(ii++A.<<\JOF(rJ   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   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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TrJ   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   rL   r   r   )r0   r:  r1   r   r/   r   r"  r   r.  rO   rq  )rG   rE   dcgdcg_ignore_ties	discountss        rH   test_dcg_tiesr,  P  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	
   rJ   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(  )r0   r1   rN   r   r   r.  )r   s    rH   test_ndcg_ignore_ties_with_kr/  d  sL    
		"f%AaaT2fmm11$/7   rJ   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)r0   r   r   r   r   r   )rG   rE   r   s      rH    test_ndcg_negative_ndarray_errorr1  k  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   rP  r   皙ɿr   r   Tr'  rY   r5  )
r0   r1   rN   r4   r5   r   r/   r   r   r.  )rG   rE   ndcgndcg_no_tiess       rH   test_ndcg_invariantr7  t  s    YYr]""1b)Fryy,,Q/77c7UUGf&Df4@L6==....6==%%%%tOGf&&--*<<<<rJ   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   rL   )r   r+   r   r4  r   r   r'  r&   r+   rP  )log_baser(  )r   r   	   rY   )r0   eyetiler1   r4   r5   r   r/   r   r   r.  r"  r   log10r   rq  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 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UrJ   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    rH   test_ndcg_error_single_documentrB    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_forr0   r4   r5   r  r  s      rH   test_ndcg_scorerE    sL    .ALIAgkG)ii++A.<<\JOF)rJ   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   r}  r   r.  r0   r   rO   r   r   r/   )rG   rE   r%  r^   all_zeros        rH   rD  rD    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<<?,,,,rJ   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+   )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_scorerL    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  r0   r   r   r   r.  )rG   r  
true_scorerE   r^   s        rH   test_top_k_accuracy_scorerS    sH     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                 *   / 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astyper0   int64r   r   r   r.  )rE   r  rR  rG   rh   rF   r^   	score_accs           rH    test_top_k_accuracy_score_binaryrZ    s     F{{})gkkmq.@aI78Avg!))"((36F A6Ev.I:z!::::::rJ   zy_true, true_score, labels)r   r+   r+   r&   r  )r   rD  rD  r   )r   r   r  rD  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  rN  rO  rP  r&   r  r  N)r0   r:  r   r   r   r.  )rG   rR  r  r[  rE   r^   s         rH   0test_top_k_accuracy_score_multiclass_with_labelsr^    s[     F#hh    		
G !AfEEFMM*----rJ   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 )NrP  r5  r   )r  r?   n_informativer*   r)   r&   r  )r   make_classificationr   r   r8   rZ  r[   r   r9   r0   r}  r   )	r=   r>   X_trainX_testy_trainy_testrD   r  scoress	            rH   $test_top_k_accuracy_score_increasingrg  ,  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  rQ  )rG   r  rR  rE   s       rH   test_top_k_accuracy_score_tiesri  ?  sC     hh		
G  15z9RRRRrJ   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+   )r0   r   r   r   r   r   )rG   r  rE   r   r^   s        rH   !test_top_k_accuracy_score_warningrk  T  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&   r]  )r   r   r   r   )rG   rE   r  rF  s       rH   test_top_k_accuracy_score_errorrm  m  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  )r0   r   r   r   r.  )r  rG   rE   results       rH   Otest_label_ranking_avg_precision_score_should_allow_csr_matrix_for_y_true_inputrp    sG     Iy12Fhh34G26CFV]]5))))rJ   metricr7  ))FT)r   r+   )r   rY   )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.
*   rP  rL   T)r   replacer  N)	r0   r4   r5   choicerK  
isinstancerP   r   any)rq  r7  rB   r?   rR   rG   y_probaro  metric_1metric_2r   s              rH   #test_ranking_metric_pos_label_typesr}    s     ))


#Cwr{yZZZ>Fhhy!GFy9F&%  88F#####)/&J88H%))++++88H%))++++88J'++-----rJ   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&   rP  r   N)r0   r4   r5   r   rK  r   isinf)r   rB   rG   rE   rt   r   s         rH   (test_roc_curve_with_probablity_estimatesr    sb     ))

 2
3C[[AB['FhhrlG 1Aq88JqM""""rJ   )NF)r   rJ  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*  r2  rB  rG  rN  rT  rW  r`  filterwarningsrf  rl  rw  rp  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/  r1  r7  r@  rB  rE  rD  rL  rS  rZ  r^  rg  ri  rk  rm  rp  r}  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:( 	,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..(#rJ   