diff --git a/research/research.py b/research/research.py index e3a68e0..3d27fb6 100644 --- a/research/research.py +++ b/research/research.py @@ -242,7 +242,9 @@ def build_categorical_features_encoder_target(*, random_state=None): # %% def build_regressor(*, random_state=None): return sklearn.ensemble.RandomForestRegressor( - 10, criterion='squared_error', max_features='sqrt', random_state=random_state, + 10, criterion='squared_error', + max_depth=8, max_features='sqrt', + random_state=random_state, )