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
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