From 3dff4ba6ecd142b2b0bb37d3bac10e598f4b28d0 Mon Sep 17 00:00:00 2001 From: Niall Turbitt Date: Wed, 29 Jun 2022 15:08:33 -0700 Subject: [PATCH] max depth update --- conf/job_configs/model_train.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/conf/job_configs/model_train.yml b/conf/job_configs/model_train.yml index dce6aed..cdae9b8 100644 --- a/conf/job_configs/model_train.yml +++ b/conf/job_configs/model_train.yml @@ -1,5 +1,5 @@ mlflow_params: - run_name: 'random_forest_baseline' + run_name: 'random_forest_max_depth_update' pipeline_params: test_size: 0.25 @@ -7,7 +7,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