From 9665ebdf4d1a8c2dff63c0b991184f633c7171f7 Mon Sep 17 00:00:00 2001 From: duiyipan Date: Sat, 14 Sep 2024 23:43:57 +0800 Subject: [PATCH] autosklearn delete per_run_time_limit and change time_limit --- expo/experimenter/autosklearn.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/expo/experimenter/autosklearn.py b/expo/experimenter/autosklearn.py index 602b8385a..e8923c6bd 100644 --- a/expo/experimenter/autosklearn.py +++ b/expo/experimenter/autosklearn.py @@ -10,7 +10,7 @@ def custom_scorer(y_true, y_pred, metric_name): class ASRunner: - time_limit = 300 + time_limit = 600 def __init__(self, state=None): self.state = state @@ -47,7 +47,6 @@ class ASRunner: if eval_metric == "rmse": automl = self.autosklearn.regression.AutoSklearnRegressor( time_left_for_this_task=self.time_limit, - per_run_time_limit=60, metric=self.create_autosklearn_scorer(eval_metric), memory_limit=8192, tmp_folder="AutosklearnModels/as-{}-{}".format( @@ -58,7 +57,6 @@ class ASRunner: elif eval_metric in ["f1", "f1 weighted"]: automl = self.autosklearn.classification.AutoSklearnClassifier( time_left_for_this_task=self.time_limit, - per_run_time_limit=60, metric=self.create_autosklearn_scorer(eval_metric), memory_limit=8192, tmp_folder="AutosklearnModels/as-{}-{}".format(