Scikit-learnでクラス数を考慮してデータを分割したい
クラス数の比率を保ったままデータを分割する

sss = StratifiedShuffleSplit()
sss.get_n_splits(X, y)