switch profiling for better perfomance

VIDA-NYU · May 1, 2024 · c519c03 · c519c03
1 parent 26e78e7
commit c519c03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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