Function cv_split

experiments/python/pyience.py:538–548

(X, y, n_folds=5, fractions=None, stratified=True)

def cv_split(X, y, n_folds=5, fractions=None, stratified=True):
    if len(X) != len(y):
        raise IndexError("len(X) {} != len(y) {}".format(len(X), len(y)))

    all_idxs = cv_partition_idxs(y, n_folds=n_folds, fractions=fractions,
                                 stratified=stratified)

    X_split = [X[idxs] for idxs in all_idxs]
    y_split = [y[idxs] for idxs in all_idxs]

    return X_split, y_split


# ================================================================ Main