model params updated

niall-turbitt · Jul 25, 2022 · 9c1bbe7 · 9c1bbe7
1 parent 238a988
commit 9c1bbe7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/conf/pipeline_configs/model_train.yml b/conf/pipeline_configs/model_train.yml
@@ -1,13 +1,13 @@
 mlflow_params:
-  run_name: 'random_forest_baseline'
+  run_name: 'random_forest_new_model'
 
 pipeline_params:
   test_size: 0.25
   random_state: 42
 
 model_params:
   n_estimators: 100
-  max_depth: 4
+  max_depth: 8
   min_samples_leaf: 1
   max_features: 'auto'
   random_state: 42