github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

check n_iter == 1

Browse Source
This commit is contained in:
Chi Wang
2021-10-16 03:09:56 -07:00
parent 46b29e05c7
commit b2d8b097d7
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_training_log.py
View File

@@ -37,10 +37,11 @@ class TestTrainingLog(unittest.TestCase):
if automl.best_estimator:
estimator, config = automl.best_estimator, automl.best_config
model0 = automl.best_model_for_estimator(estimator)
print(model0.estimator)
print(model0.params["n_estimators"], model0.estimator)
automl.time_budget = None
model, _ = automl._state._train_with_config(estimator, config)
print(model.estimator)
# model0 and model are equivalent unless model0's n_estimator is out of search space range
assert (
str(model0.estimator) == str(model.estimator)