Update hyperparameter tuningx

bodywork-ml · Jun 20, 2022 · 463298e · 463298e
1 parent d7e4e92
commit 463298e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pipeline/train_model.py b/pipeline/train_model.py
@@ -19,9 +19,9 @@
 HYPERPARAM_GRID = {
     "random_state": [42],
     "criterion": ["squared_error", "absolute_error"],
-    "max_depth": [2, 3, 4, 5, 6, 7, 8, 9, 10, None],
-    "min_samples_split": [2, 3, 4, 5, 6, 7, 8, 9, 10],
-    "min_samples_leaf": [2, 3, 4, 5, 6, 7, 8, 9, 10],
+    "max_depth": [2, 4, 6, 8, 10, None],
+    "min_samples_split": [2, 4, 6, 8, 10],
+    "min_samples_leaf": [2, 4, 6, 8, 10],
 }
 
 log = logging.configure_logger()