random_split

random_split(y, num_train_per_class=20, num_val=500)[source]