コード例 #1
0
ファイル: randomforest.py プロジェクト: hdemers/datascience
def cv_max_features(clf, data, max_features, predictors, target, cv=5):
    mses = []
    n_features = range(1, max_features)
    for n_feature in n_features:
        clf.max_features = n_feature
        mses.append(cv_mse(clf, data, predictors, target, cv))
    return pd.DataFrame({'mse': mses}, index=n_features)
コード例 #2
0
ファイル: randomforest.py プロジェクト: hdemers/datascience
def cv_n_estimators(clf, data, max_estimators, predictors, target,
                    cv=5, step=1):
    mses = []
    n_estimators = range(1, max_estimators, step)
    for n_estimator in n_estimators:
        clf.n_estimators = n_estimator
        mses.append(cv_mse(clf, data, predictors, target, cv))
    return pd.DataFrame({'mse': mses}, index=n_estimators)