debug

VIDA-NYU · Apr 17, 2024 · 8eb424c · 8eb424c
1 parent 162bd45
commit 8eb424c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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')