Merge pull request #45 from niall-turbitt/dev/new_model

max depth update
niall-turbitt · Jun 29, 2022 · 86c8a67 · 86c8a67
2 parents 192f53b + 3dff4ba
commit 86c8a67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/conf/job_configs/model_train.yml b/conf/job_configs/model_train.yml
@@ -1,13 +1,13 @@
 mlflow_params:
-  run_name: 'random_forest_baseline'
+  run_name: 'random_forest_max_depth_update'
 
 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