github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-02-16 18:45:19 -05:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Anonymous-submission-repo
2022-10-14 20:40:49 +00:00
parent 2334a9c81d
commit bf81912f09
11 changed files with 131 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/automl/test_lexiflow.py
View File

@@ -24,7 +24,10 @@ def test_lexiflow():
"mem_thres": 128 * (1024**3),
}
automl.fit(X_train=X_train, y_train=y_train, X_val=X_test, y_val=y_test, **settings)
print(automl.predict(X_train))
print(automl.model)
print(automl.config_history)
print(automl.best_iteration)
print(automl.best_estimator)
if __name__ == "__main__":
test_lexiflow()

24
test/tune/test_lexiflow.py
View File

@@ -109,15 +109,35 @@ def test_lexiflow():
"n_epoch": 1,
}
tune.run(
# lexico tune
analysis = tune.run(
evaluate_function,
num_samples=-1,
time_budget_s=100,
time_budget_s=50,
config=search_space,
use_ray=False,
lexico_objectives=lexico_objectives,
low_cost_partial_config=low_cost_partial_config,
)
print(analysis.best_trial)
print(analysis.best_config)
print(analysis.best_result)
# Non lexico tune
analysis = tune.run(
evaluate_function,
metric = "error_rate",
mode = "min",
num_samples=-1,
time_budget_s=50,
config=search_space,
use_ray=False,
lexico_objectives=None,
low_cost_partial_config=low_cost_partial_config,
)
print(analysis.best_trial)
print(analysis.best_config)
print(analysis.best_result)
if __name__ == "__main__":