Merge pull request #8 from aiwithqasim/dev/max_depth_update

updated max_params
aiwithqasim · Feb 21, 2023 · a3bb2ee · a3bb2ee
2 parents dfd4670 + c6ef4d6
commit a3bb2ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/conf/pipeline_configs/model_train.yml b/conf/pipeline_configs/model_train.yml
@@ -1,13 +1,13 @@
 mlflow_params:
-  run_name: 'deep_forest_mini'
+  run_name: 'deep_forest'
 
 pipeline_params:
   test_size: 0.33
   random_state: 42
 
 model_params:
   n_estimators: 150
-  max_depth: 8
+  max_depth: 12
   min_samples_leaf: 1
   max_features: 'auto'
   random_state: 206