From 9c1bbe7338109029ceeedf4a33663410faa15b19 Mon Sep 17 00:00:00 2001 From: Niall Turbitt Date: Mon, 25 Jul 2022 12:38:14 +0100 Subject: [PATCH] model params updated --- conf/pipeline_configs/model_train.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/conf/pipeline_configs/model_train.yml b/conf/pipeline_configs/model_train.yml index dce6aed..2a3b7e5 100644 --- a/conf/pipeline_configs/model_train.yml +++ b/conf/pipeline_configs/model_train.yml @@ -1,5 +1,5 @@ mlflow_params: - run_name: 'random_forest_baseline' + run_name: 'random_forest_new_model' 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