From 801c40a0bc8ce6f43f1973c1745165c20fbcc676 Mon Sep 17 00:00:00 2001 From: EdenWuyifan Date: Mon, 14 Aug 2023 02:01:43 -0400 Subject: [PATCH] tmp --- alpha_automl/automl_api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/alpha_automl/automl_api.py b/alpha_automl/automl_api.py index 81fa3341..28918e2f 100644 --- a/alpha_automl/automl_api.py +++ b/alpha_automl/automl_api.py @@ -133,8 +133,9 @@ def fit(self, X, y): leaderboard_data.append([index, pipeline.get_summary(), pipeline.get_score()]) self.leaderboard = pd.DataFrame(leaderboard_data, columns=['ranking', 'pipeline', self.metric]) - + best_pipeline_id = PIPELINE_PREFIX + '1' + logger.critical(f'[yfw] {self.pipelines[best_pipeline_id].get_pipeline()}') self._fit(X, y, best_pipeline_id) def predict(self, X):