From 574f1b0e0d2c6b9702097287761bf64a29b8a82f Mon Sep 17 00:00:00 2001 From: duiyipan Date: Sun, 15 Sep 2024 00:01:35 +0800 Subject: [PATCH] change import way --- expo/experimenter/autosklearn.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/expo/experimenter/autosklearn.py b/expo/experimenter/autosklearn.py index c6aa70920..9d0ea2df4 100644 --- a/expo/experimenter/autosklearn.py +++ b/expo/experimenter/autosklearn.py @@ -24,6 +24,8 @@ class ASRunner: ) def run(self): + import autosklearn + train_path = self.datasets["train"] dev_wo_target_path = self.datasets["dev_wo_target"] test_wo_target_path = self.datasets["test_wo_target"] @@ -37,9 +39,7 @@ class ASRunner: y_train = train_data[target_col] if eval_metric == "rmse": - from autosklearn.regression import AutoSklearnRegressor - - automl = AutoSklearnRegressor( + automl = autosklearn.regression.AutoSklearnRegressor( time_left_for_this_task=self.time_limit, metric=self.create_autosklearn_scorer(eval_metric), memory_limit=8192, @@ -49,9 +49,7 @@ class ASRunner: n_jobs=-1, ) elif eval_metric in ["f1", "f1 weighted"]: - from autosklearn.classification import AutoSklearnClassifier - - automl = AutoSklearnClassifier( + automl = autosklearn.classification.AutoSklearnClassifier( time_left_for_this_task=self.time_limit, metric=self.create_autosklearn_scorer(eval_metric), memory_limit=8192,