diff --git a/pipeline/train_model.py b/pipeline/train_model.py index 1fcaf13..5ff53d4 100644 --- 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()