From c519c03b05e3eadeae20c4ce204fb5268556dc4f Mon Sep 17 00:00:00 2001 From: EdenWuyifan Date: Wed, 1 May 2024 00:48:32 -0400 Subject: [PATCH] switch profiling for better perfomance --- alpha_automl/automl_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/alpha_automl/automl_manager.py b/alpha_automl/automl_manager.py index c950c29c..b3fe9bb2 100644 --- a/alpha_automl/automl_manager.py +++ b/alpha_automl/automl_manager.py @@ -52,8 +52,8 @@ def search_pipelines(self, X, y, scoring, splitting_strategy, automl_hyperparams def _search_pipelines(self, automl_hyperparams): search_start_time = time.time() automl_hyperparams = self.check_automl_hyperparams(automl_hyperparams) - metadata = profile_data(self.X) X, y, is_sample = sample_dataset(self.X, self.y, SAMPLE_SIZE, self.task) + metadata = profile_data(X) internal_splitting_strategy = make_splitter(SPLITTING_STRATEGY) self.found_pipelines = 0 need_rescoring = True