hSSKrSrSSjrg)NcX-UR5- n[R"U5n[XR5- 5nUS:aX4- n[R"[R "U55SSS2nUHZn[R "Xh:H5un [[U 5U5n URXSS9n XI==S- ss'XZ-nUS:XdMZ O UR[R5$)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 Q/home/james-whalen/.local/lib/python3.13/site-packages/datasets/utils/stratify.pyapproximate_moder s0',*:*:*<>"(( ##c #2# [R"USS9upVURSn[R"U5n[R"U5S:a [ S5eX:a[ SX4-5eX':a[ SX'4-5e[R "[R"USS 9[R"U5S S 5n [U5Hn [XU5n X- n [XU5n /n/n[U5HanURUU5nU URUS S 9nURUS U U5 URUU UU UU U-5 Mc URU5nURU5nX4v M g 7f)aP 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_itraintestir0perm_indices_class_is r)stratified_shuffle_split_generate_indicesrC6s.1T:G a I;;y)L vvla455 Z^e]q q   Y]c\o o  HHRZZ DbiiP\F]^a_aFbcM 8_|c:!-!35sCy!A//,q/:K#0#3#8#86#8#R LL-hA7 8 KK,SVc!fs1voF G " &t$k!sFF) )numpyr r rCr!rrGs/$d5r!