apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-11 15:15:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

autosklearn delete per_run_time_limit and change time_limit

Browse source
This commit is contained in:
duiyipan 2024-09-14 23:43:57 +08:00
parent c4fe056bca
commit 9665ebdf4d
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
expo/experimenter/autosklearn.py
View file

@ -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(