+gdddlZdZddZdS)Nc:||z|z }tj|}t||z }|dkr||z }tjtj|ddd}|D]n}tj||k\} tt| |} | | | d} || xxdz cc<|| z}|dkrno| tj S)aComputes 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 rNF)sizereplace) sumnpfloorintsortuniquewhereminlenchoiceastypeint64) class_countsn_drawsrng continuousfloored need_to_add remaindervaluesvalueindsadd_nows 7lib/python3.11/site-packages/datasets/utils/stratify.pyapproximate_moder s0<',*:*:*<*<>"( # ## c #Ktj|d\}}|jd}tj|}tj|dkrt d||krt d||fz||krt d||fztjtj|d tj|d d } t|D]} t|||} || z } t| ||} g}g}t|D]}| ||}| | |d }| |d | || || || || |z| |}| |}||fVd S)a Provides train/test indices to split data in train/test sets. It's reference is taken from StratifiedShuffleSplit implementation of scikit-learn library. Args ---------- n_train : int, represents the absolute number of train samples. n_test : int, represents the absolute number of test samples. random_state : int or RandomState instance, default=None Controls the randomness of the training and testing indices produced. Pass an int for reproducible output across multiple function calls. n_splits : int, default=10 Number of re-shuffling & splitting iterations. T)return_inverserzMinimum class count errorzLThe train_size = %d should be greater or equal to the number of classes = %dzKThe test_size = %d should be greater or equal to the number of classes = %d mergesort)kindNrclip)mode)r r shapebincountr ValueErrorsplitargsortcumsumranger permutationtakeextend)yn_trainn_testrn_splitsclasses y_indices n_classesr class_indices_n_iclass_counts_remainingt_itraintestir1perm_indices_class_is r)stratified_shuffle_split_generate_indicesrD6s.1T:::GY a I;y))L vla4555 ]ahjs`t t     \`fhq_r r   HRZ DDDbiP\F]F]^a_a^aFbccM 8__|Wc::!-!35vsCCy!! H HA//,q/::K#0#3#8#86#8#R#R LL-hAh7 8 8 8 KK,SVc!fs1vo-EF G G G G&&t$$Tk!r!)r")numpyr r rDr!rrGs?/$/$/$d555555r!