diff --git a/train.py b/train.py index 92678f346..43b3a7873 100644 --- a/train.py +++ b/train.py @@ -23,7 +23,7 @@ ################################# # Fit a model on the train section -regr = RandomForestRegressor(max_depth=5, random_state=seed) +regr = RandomForestRegressor(max_depth=40, random_state=seed) regr.fit(X_train, y_train) # Report training set score