debug score_pipeline

VIDA-NYU · Jul 22, 2024 · 4d9d8d6 · 4d9d8d6
1 parent f74729c
commit 4d9d8d6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/alpha_automl/automl_api.py b/alpha_automl/automl_api.py
@@ -123,10 +123,10 @@ def fit(self, X, y):
             # [SMAC] added here!!
             if self.optimizing and index <= 10:
                 opt_pipeline = optimizer.optimize_pipeline(pipeline.get_pipeline())
-                opt_score, _, _ = score_pipeline(opt_pipeline, X, y, self.scorer, self.splitter)
-                logger.critical(f'[SMAC] {pipeline_id} successfully optimized: {pipeline.get_score()} => {opt_score}')
-                pipeline.set_pipeline(opt_pipeline)
-                pipeline.set_score(opt_score)
+                smac_pipeline = score_pipeline(opt_pipeline, X, y, self.scorer, self.splitter)
+                if smac_pipeline.get_score() > pipeline.get_score():
+                    logger.critical(f'[SMAC] {pipeline_id} successfully optimized: {pipeline.get_score()} => {smac_pipeline.get_score()}')
+                    pipeline = smac_pipeline
             leaderboard_data.append([index, pipeline.get_summary(), pipeline.get_score()])
 
         self.leaderboard = pd.DataFrame(leaderboard_data, columns=['ranking', 'pipeline', self.metric])