Beispiel #1
0
def fit_all(**rf_params):
    regr = RandomForestRegressor(**rf_params)
    # Make sure all cores are used.
    regr.n_jobs = get_ncpus()
    regr.fit(all_splits.X_train, all_splits.y_train)
    return regr
Beispiel #2
0
def fit_rf_out_season(**rf_params):
    regr = RandomForestRegressor(**rf_params)
    # Make sure all cores are used.
    regr.n_jobs = get_ncpus()
    regr.fit(out_fs_splits.X_train, out_fs_splits.y_train)
    return regr