def METHOD_NAME():
    train_x, train_y = _load_data()
    print('loading data done!')
    folds = list(
        StratifiedKFold(
            n_splits=10, shuffle=True, random_state=args.random_seed).split(
                train_x, train_y))
    fold_index = []
    for i, (train_id, valid_id) in enumerate(folds):
        fold_index.append(valid_id)
    print("fold num: %d" % (len(fold_index)))
    fold_index = np.array(fold_index)
    np.save(
        os.path.join(args.src_path, 'fold_index.npy'),
        fold_index,
        allow_pickle=True)
    save_x_y(fold_index, train_x, train_y)
    print("save train_x_y done!")
    fold_index = np.load(
        os.path.join(args.src_path, "fold_index.npy"), allow_pickle=True)
    save_i(fold_index)
    print("save index done!")