
     h	                     T    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 ddl
mZ 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).    N)partial)check_return_X_y)assert_almost_equal)assert_array_equalc                 t    | d          }|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
                            d	          sJ g d
}t          ||d d                    d}d}t          ||          D ]6\  }	}
|                    |
          }|	|d d |f         j         j        k    sJ 7 | ddd          }|j         |j        }}|j        }t          | dd          }t!          ||           t          t#          j        |d d                   t#          j        |                     d}|D ]}|                                                    |          }|                                                    |          }||d d f                                         }||d d f                                         }t)          ||           ||d d f                                         }||d d f                                         }t)          ||           d S )NF)shufflei}i' )>F i  )r	   g   )r	   r
   z.. _rcv1_dataset:)C11C12C13C14C15C151   )GMILE143CCAT)   i  i TtrainM   )r   subsetrandom_state)r   r   imZ  )i  i  i6  )datatargettarget_namestolist	sample_idspissparsesizeshapelenDESCR
startswithr   zipindexr   r   npsorttoarrayr   )fetch_rcv1_fxt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/html/Sam_Eipo/venv/lib/python3.11/site-packages/sklearn/datasets/tests/test_rcv1.pytest_fetch_rcv1rD      s   N5)))EZB%,,..bH ;r???;r???rw|####bgl"""" bh&&&&BH$$$$    #h-- ;!!"566666 CBB'"1"666 /O..@@ ) )SNN3bAhm(((((( N4bIIIEZB	B wGGGJUJ''' rwr&5&z**BGBKK888 *O$ 
> 
>	yy{{  ++yy{{  ++dAAAg;..00dAAAg;..00,.>???T111W+--//T111W+--//O_====
> 
>    )__doc__scipy.sparsesparser   numpyr(   	functoolsr   "sklearn.datasets.tests.test_commonr   sklearn.utils._testingr   r   rD    rE   rC   <module>rN      s                     ? ? ? ? ? ? 6 6 6 6 6 6 5 5 5 5 5 57> 7> 7> 7> 7>rE   