early

Author: richardliaw

examples/bohb_example.py

@@ -22,8 +22,9 @@
 bohb_tune_search = TuneSearchCV(
     SGDClassifier(),
     param_distributions=param_dists,
-    n_iter=2,
+    n_iter=20,
     max_iters=10,
+    verbose=2,
     search_optimization="bohb")
 
 bohb_tune_search.fit(X_train, y_train)

tune_sklearn/tune_basesearch.py

@@ -256,8 +256,9 @@ def __init__(self,
                 # the next block
                 early_stopping = "AsyncHyperBandScheduler"
             # Resolve the early stopping object
-            self.early_stopping = resolve_early_stopping(
+            early_stopping = resolve_early_stopping(
                 early_stopping, self.max_iters)
+        self.early_stopping = early_stopping
 
         self.cv = cv
         self.scoring = scoring