feature_engineering.py 文件源码

def CV_eval(model, X, y):
    '''
    Perform 8-fold cross-validation

    Input: model, X data, Y data
    Return: mean of cross-val accuracy scores
    '''
    scores = cross_val_score(model, X, y, cv=8)
    pprint (scores)
    print("Accuracy: %0.2f (+/- %0.2f)" % (scores.mean(), scores.std() * 2))
    return scores.mean()