From fad9450edf62a5472f5135f4ae10d6d0a7fb3ddf Mon Sep 17 00:00:00 2001 From: Eden Wu Date: Sun, 1 Sep 2024 18:45:17 -0400 Subject: [PATCH] temp --- alpha_automl/automl_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/alpha_automl/automl_api.py b/alpha_automl/automl_api.py index 4b17721..144e9ec 100644 --- a/alpha_automl/automl_api.py +++ b/alpha_automl/automl_api.py @@ -121,7 +121,7 @@ def fit(self, X, y): pipeline_id = PIPELINE_PREFIX + str(index+1) self.pipelines[pipeline_id] = pipeline # [SMAC] added here!! - if self.optimizing and index < 10: + if self.optimizing and index < 5: opt_pipeline = optimizer.optimize_pipeline(pipeline.get_pipeline()) smac_pipeline = score_pipeline(opt_pipeline, X_sample, y_sample, self.scorer, self.splitter, self.task_type) if smac_pipeline.get_score() < pipeline.get_score():