From b80fbde038785b4a883a99f74dedf84a2986a9cd Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Thu, 10 Sep 2015 02:21:54 +0000 Subject: [PATCH] change defaults --- model/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/model.py b/model/model.py index 6b37c7d..c17f7f9 100644 --- a/model/model.py +++ b/model/model.py @@ -27,7 +27,7 @@ def fit_regressor(X, y, window): ''' Fits regressor model using cross validation ''' - model = RandomForestRegressor(n_estimators=50, + model = RandomForestRegressor(n_estimators=100, min_samples_leaf=500, random_state=42, n_jobs=-1)