
    0Ph'	                     L    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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                 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|          }|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train)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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/html/test/jupyter/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(((((( NW3E  E 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__	functoolsr   numpyr'   scipy.sparsesparser   "sklearn.datasets.tests.test_commonr   sklearn.utils._testingr   r   rD    rE   rC   <module>rN      s                     ? ? ? ? ? ? J J J J J J J J9> 9> 9> 9> 9>rE   