diff --git a/conf/pipeline_configs/model_train.yml b/conf/pipeline_configs/model_train.yml index 141f9be..2ae9c31 100644 --- a/conf/pipeline_configs/model_train.yml +++ b/conf/pipeline_configs/model_train.yml @@ -8,7 +8,7 @@ pipeline_params: model_params: n_estimators: 100 - max_depth: 4 + max_depth: 8 min_samples_leaf: 1 max_features: 'auto' random_state: 42 \ No newline at end of file