Ë
    ÷Q(h'	  ã                   óJ   — d Z ddlmZ ddlZddlmZ ddlm	Z	 ddl
mZmZ d„ Zy)zyTest the rcv1 loader, if the data is available,
or if specifically requested via environment variable
(e.g. for CI jobs).é    )ÚpartialN)Úcheck_return_X_y)Úassert_almost_equalÚassert_array_equalc                 óH  —  | d¬«      }|j                   |j                  }}|j                  j                  «       |j                  }}t        j                  |«      sJ ‚t        j                  |«      sJ ‚d|j                   j                  k(  sJ ‚d|j                   j                  k(  sJ ‚d|j                  k(  sJ ‚d|j                  k(  sJ ‚d|j                  k(  sJ ‚dt        |«      k(  sJ ‚|j                  j                  d	«      sJ ‚g d
¢}t        ||d d «       d}d}	t        |	|«      D ]8  \  }
}|j                  |«      }|
|d d …|f   j                   j                  k(  rŒ8J ‚  | dd|¬«      }|j                   |j                  }}|j                  }t        | dd¬«      }t!        ||«       t        t#        j$                  |d d «      t#        j$                  |«      «       d}|D ]´  }|j                  «       j                  |«      }|j                  «       j                  |«      }||d d …f   j'                  «       }||d d …f   j'                  «       }t)        ||«       ||d d …f   j'                  «       }||d d …f   j'                  «       }t)        ||«       Œ¶ y )NF)Úshufflei©}¡iÇ' )é>F i„¸  )r	   ég   )r	   r
   z.. _rcv1_dataset:)ÚC11ÚC12ÚC13ÚC14ÚC15ÚC151é   )ÚGMILÚE143ÚCCAT)é   i¶  i�Ñ TÚtrain)r   ÚsubsetÚrandom_state)r   r   imZ  )iî  iÊ  iÚ6  )ÚdataÚtargetÚtarget_namesÚtolistÚ	sample_idÚspÚissparseÚsizeÚshapeÚlenÚDESCRÚ
startswithr   ÚzipÚindexr   r   ÚnpÚsortÚtoarrayr   )Úfetch_rcv1_fxtÚglobal_random_seedÚdata1ÚX1ÚY1Úcat_listÚs1Úfirst_categoriesÚsome_categoriesÚnumber_non_zero_in_catÚnumÚcatÚjÚdata2ÚX2ÚY2Ús2Ú
fetch_funcÚsome_sample_idsr   Úidx1Úidx2Úfeature_values_1Úfeature_values_2Útarget_values_1Útarget_values_2s                             ú^/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sklearn/datasets/tests/test_rcv1.pyÚtest_fetch_rcv1rD      s{  € Ù 5Ô)€EØ�Z‰Z˜Ÿ™ˆ€BØ×%Ñ%×,Ñ,Ó.°·±ˆb€Hô �;‰;�rŒ?Ðˆ?Ü�;‰;�rŒ?Ðˆ?Ø�r—w‘w—|‘|Ò#Ð#Ð#Ø�b—g‘g—l‘lÒ"Ð"Ð"ð ˜bŸh™hÒ&Ð&Ð&Ø˜BŸH™HÒ$Ð$Ð$Ø˜Ÿ™Ò Ð Ð Ø”#�h“-ÒÐÐð �;‰;×!Ñ!Ð"5Ô6Ð6Ð6ò CÐÜÐ'¨°"°1¨Ô6ð /€OØ.ÐÜÐ.°Ó@ò )‰ˆˆSØ�N‰N˜3ÓˆØ�bš˜A˜‘h—m‘m×(Ñ(Ó(Ð(Ð(ð)ñ
 Ø˜WÐ3Eô€Eð �Z‰Z˜Ÿ™ˆ€BØ	�‰€Bô ˜°¸wÔG€JÜ�U˜JÔ'ô ”r—w‘w˜r & 5˜zÓ*¬B¯G©G°B«KÔ8ð *€OØ$ò 
>ˆ	Ø�y‰y‹{× Ñ  Ó+ˆØ�y‰y‹{× Ñ  Ó+ˆà˜d¢A˜g™;×.Ñ.Ó0ÐØ˜d¢A˜g™;×.Ñ.Ó0ÐÜÐ,Ð.>Ô?à˜T¢1˜W™+×-Ñ-Ó/ˆØ˜T¢1˜W™+×-Ñ-Ó/ˆÜ˜O¨_Õ=ñ
>ó    )Ú__doc__Ú	functoolsr   Únumpyr'   Úscipy.sparseÚsparser   Ú"sklearn.datasets.tests.test_commonr   Úsklearn.utils._testingr   r   rD   © rE   rC   ú<module>rN      s    ðñõ ã Ý å ?ß Jó9>rE   