build_model.py 文件源码

def split_testing_data_r(y):
    ss = ShuffleSplit(n_splits=1, test_size=0.2)
    tri = None
    tei = None
    for itr, ite in ss.split(y):
        tri = itr
        tei = ite
    return tri, tei