Merge pull request #9 from thuijskens/feature/scikit-update

Update scikit-learn to 0.21
thuijskens · Nov 2, 2019 · 68bef5a · 68bef5a
2 parents 89398ba + 05af0c3
commit 68bef5a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hyperband/search.py b/hyperband/search.py
@@ -335,8 +335,8 @@ def __init__(self, estimator, param_distributions,
         self.random_state = random_state
 
         super(HyperbandSearchCV, self).__init__(
-            estimator=estimator, scoring=scoring, fit_params=None,
-            n_jobs=n_jobs, iid=iid, refit=refit, cv=cv, verbose=verbose,
+            estimator=estimator, scoring=scoring, n_jobs=n_jobs,
+            iid=iid, refit=refit, cv=cv, verbose=verbose,
             pre_dispatch=pre_dispatch, error_score=error_score,
             return_train_score=return_train_score)
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 nose>=1.1.2
-scikit-learn>=0.19
-numpy>=1.8.0
+scikit-learn>=0.21
+numpy>=1.11.0