mirror of
https://github.com/microsoft/autogen.git
synced 2026-02-16 23:25:55 -05:00
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:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user