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

new model params
niall-turbitt · Jun 29, 2022 · 41e59ea · 41e59ea
2 parents 5d2d45a + 65ee13a
commit 41e59ea
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_new_max_depth'
 
 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