separate search alg and schedulers from recipe (intel-analytics#3041)

* seperate search alg and schedulers from fixed recipe * add corresponding docs and expose to zouwu AutoTrainer
shanyu-sys · Nov 9, 2020 · 4ef8880 · 4ef8880
1 parent d2f0cda
commit 4ef8880
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/python/orca/example/automl/AutoXGBoostClassifier.py b/python/orca/example/automl/AutoXGBoostClassifier.py
@@ -114,7 +114,17 @@
 
     estimator = AutoXGBoost().classifier(feature_cols=input_cols,
                                          target_col="ArrDelayBinary",
-                                         config=config)
+                                         config=config,
+                                         search_alg="skopt",
+                                         search_alg_params=None,
+                                         scheduler="AsyncHyperBand",
+                                         scheduler_params=dict(
+                                             max_t=50,
+                                             grace_period=1,
+                                             reduction_factor=3,
+                                             brackets=3,
+                                         ),
+                                         )
 
     import time
     start = time.time()