From 8eb424c2aeb33e99966501147205c710260cb71c Mon Sep 17 00:00:00 2001 From: EdenWuyifan Date: Wed, 17 Apr 2024 16:12:21 -0400 Subject: [PATCH] debug --- alpha_automl/automl_manager.py | 3 ++- alpha_automl/pipeline_synthesis/setup_search.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/alpha_automl/automl_manager.py b/alpha_automl/automl_manager.py index d285c11a..4bb7086a 100644 --- a/alpha_automl/automl_manager.py +++ b/alpha_automl/automl_manager.py @@ -66,7 +66,8 @@ def _search_pipelines(self, automl_hyperparams): found_pipelines = 0 - while pipelines: + pipeline_threshold = 20 + while pipelines and found_pipelines < pipeline_threshold: pipeline = pipelines.pop() try: alphaautoml_pipeline = score_pipeline(pipeline, self.X, self.y, self.scoring, diff --git a/alpha_automl/pipeline_synthesis/setup_search.py b/alpha_automl/pipeline_synthesis/setup_search.py index c1aab1b7..a2f37ef4 100644 --- a/alpha_automl/pipeline_synthesis/setup_search.py +++ b/alpha_automl/pipeline_synthesis/setup_search.py @@ -167,8 +167,9 @@ def evaluate_pipeline(primitives, origin): pipeline_search_rllib(game, time_bound, save_checkpoint=save_checkpoint) logger.debug('Search completed') - - return read_result_to_pipeline(builder, output_folder)[:20] + + results = read_result_to_pipeline(builder, output_folder) + return results # queue.put('DONE')