apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-03 04:42:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

change import way

Browse source
This commit is contained in:
duiyipan 2024-09-15 00:01:35 +08:00
parent c007d0bd5e
commit 574f1b0e0d
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
expo/experimenter/autosklearn.py
View file

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