tweak gb

alan-turing-institute · Oct 10, 2024 · 70b974b · 70b974b
1 parent 934ec4b
commit 70b974b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autoemulate/emulators/gradient_boosting.py b/autoemulate/emulators/gradient_boosting.py
@@ -109,7 +109,7 @@ def get_grid_params(self, search_type="random"):
             "min_samples_leaf": randint(1, 6),
             "subsample": uniform(0.6, 0.4),  # 0.4 is the range width (1.0 - 0.6)
             "max_features": ["sqrt", "log2", None],
-            "ccp_alpha": loguniform(0.01, 0.1),
+            "ccp_alpha": loguniform(0.001, 0.1),
         }
 
         param_space_bayes = {