diff --git a/conf/job_configs/model_train.yml b/conf/job_configs/model_train.yml index dce6aed..045fa73 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_new_max_depth' 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