RandomForest.py 文件源码

def crossV(model, X, y, folds = 5):

    rmse = make_scorer(RMSE, greater_is_better = False)

    scores = cross_val_score(model, X, y, cv = folds, scoring=rmse, n_jobs = 1)

    print scores
    print("Accuracy: %0.2f (+/- %0.2f)" % (scores.mean(), scores.std() * 2))