Skip transform (#665)

* Skip transform

* Fix logic and docstring, add test

* Add period ending to skip_transform doc

* Add skip_transform to retrain_from_log method

* Update test/automl/test_classification.py

Co-authored-by: Xueqing Liu <liususan091219@users.noreply.github.com>

Co-authored-by: Xueqing Liu <liususan091219@users.noreply.github.com>
This commit is contained in:
jmrichardson
2022-08-11 19:41:23 -04:00
committed by GitHub
parent b6e8b9ccca
commit 25ad397d55
2 changed files with 29 additions and 1 deletions

View File

@@ -155,6 +155,25 @@ class TestClassification(unittest.TestCase):
# "verbose": 4,
"ensemble": True,
}
automl_settings["keep_search_state"] = True
automl.fit(X, y, **automl_settings)
X, y = automl._X_train_all, automl._y_train_all
del automl
automl = AutoML()
automl_settings = {
"time_budget": 3,
"task": "classification",
"n_jobs": 1,
"estimator_list": ["kneighbor"],
"eval_method": "cv",
"n_splits": 3,
"metric": "accuracy",
"log_training_metric": True,
# "verbose": 4,
"ensemble": True,
"skip_transform": True,
}
automl.fit(X, y, **automl_settings)
del automl