repeated_cv.py

# Authors:  Denis A. Engemann <[email protected]>
#
# License:  simplified BSD 3 clause

from sklearn.cross_validation import StratifiedKFold


def repeated_folds(y, n_folds, n_repeats, seeds):
    use_seeds = seeds[:n_repeats]
    cv = [list(  # resample
        StratifiedKFold(y=y, n_folds=n_folds, shuffle=True, random_state=seed))
        for seed in use_seeds]
    cv = sum(cv, list())  # concatenate
    return cv