build_model.py 文件源码

def split_testing_data_c(y):
    sss = StratifiedShuffleSplit(n_splits=1, test_size=0.2)
    tri = None
    tei = None
    for itr, ite in sss.split(np.zeros(len(y)), y):
        tri = itr
        tei = ite
    return tri, tei