a
    �ù»jn	  ã                   @   sL   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d„ ZdS )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                 C   sø  | dd�}|j |j }}|j ¡ |j }}t |¡s8J ‚t |¡sFJ ‚d|j jksVJ ‚d|j jksfJ ‚d|jkstJ ‚d|jks‚J ‚d|jks�J ‚dt	|ƒks J ‚|j
 d	¡s°J ‚g d
¢}t||d d… ƒ d}d}	t|	|ƒD ].\}
}| |¡}|
|d d …|f j jksÜJ ‚qÜ| dd|d�}|j |j }}|j}t| ddd�}t||ƒ tt |d d… ¡t |¡ƒ d}|D ]†}| ¡  |¡}| ¡  |¡}||d d …f  ¡ }||d d …f  ¡ }t||ƒ ||d d …f  ¡ }||d d …f  ¡ }t||ƒ �qld S )NF)Úshufflei©}¡iÇ' )é>F i„¸  )r   ég   )r   r   z.. _rcv1_dataset:)ZC11ZC12ZC13ZC14ZC15ZC151é   )ZGMILZE143ZCCAT)é   i¶  i�Ñ TÚtrain)r   ÚsubsetZrandom_state)r   r   imZ  )iî  iÊ  iÚ6  )ÚdataÚtargetZtarget_namesÚtolistÚ	sample_idÚspÚissparseÚsizeÚshapeÚlenZDESCRÚ
startswithr   ÚzipÚindexr   r   ÚnpÚsortZtoarrayr   )Zfetch_rcv1_fxtZglobal_random_seedZdata1ZX1ZY1Zcat_listÚs1Zfirst_categoriesZsome_categoriesZnumber_non_zero_in_catÚnumÚcatÚjZdata2ZX2ZY2Ús2Z
fetch_funcZsome_sample_idsr   Zidx1Zidx2Zfeature_values_1Zfeature_values_2Ztarget_values_1Ztarget_values_2© r    úFE:\soccer\ml_env\Lib\site-packages\sklearn/datasets/tests/test_rcv1.pyÚtest_fetch_rcv1   sJ    

ÿ

r"   )Ú__doc__Ú	functoolsr   Únumpyr   Zscipy.sparseÚsparser   Z"sklearn.datasets.tests.test_commonr   Zsklearn.utils._testingr   r   r"   r    r    r    r!   Ú<module>   s   