Name	Size	Mode	Actions
beam_utils.cpython-36.pyc	2017	0644	editdlrm
deprecation_utils.cpython-36.pyc	3347	0644	editdlrm
doc_utils.cpython-36.pyc	654	0644	editdlrm
download_manager.cpython-36.pyc	156	0644	editdlrm
extract.cpython-36.pyc	10868	0644	editdlrm
filelock.cpython-36.pyc	10484	0644	editdlrm
file_utils.cpython-36.pyc	17904	0644	editdlrm
info_utils.cpython-36.pyc	4536	0644	editdlrm
logging.cpython-36.pyc	7258	0644	editdlrm
metadata.cpython-36.pyc	17940	0644	editdlrm
patching.cpython-36.pyc	3438	0644	editdlrm
py_utils.cpython-36.pyc	26316	0644	editdlrm
readme.cpython-36.pyc	8900	0644	editdlrm
stratify.cpython-36.pyc	3129	0644	editdlrm
tf_utils.cpython-36.pyc	1800	0644	editdlrm
typing.cpython-36.pyc	355	0644	editdlrm
version.cpython-36.pyc	4539	0644	editdlrm
_hf_hub_fixes.cpython-36.pyc	3263	0644	editdlrm
__init__.cpython-36.pyc	381	0644	editdlrm

3 <%Eg��@sdd
lZ
dd�Zddd�
Zd
S)�NcCs�|
||j�}t
j|�
}t|
|j��
}|d
kr�||}t
jt
j|�
�
ddd�}x\|D]T}t
j||k�
\
} tt| �
|�} |j | | dd�} || d7<|| 8}|d
krZPqZW|j t
j�
S)a�Computes approximate mode of multivariate hypergeometric. This is an approximation to the mode of the multivariate hypergeometric given by class_counts and n_draws. It shouldn't be off by more than one. It is the mostly likely outcome of drawing n_draws many samples from the population given by class_counts. Args ---------- class_counts : ndarray of int Population per class. n_draws : int Number of draws (samples to draw) from the overall population. rng : random state Used to break ties. Returns ------- sampled_classes : ndarray of int Number of samples drawn from each class. np.sum(sampled_classes) == n_draws r
N�
F)�size�replace�����)�sum�np�floor�int�sort�unique�where�min�len�choiceZastypeZint64)�class_countsZn_draws�rngZ continuousZflooredZneed_to_add� remainder�values�valueZindsZadd_now�r�A/usr/local/lib/python3.6/site-packages/datasets/utils/stratify.py�approximate_modes 









r� ccsV
tj
|d
d�\}}|jd}tj|�
}tj|�
dkr
s2