Skip to content

Commit 765f36a

Browse files
committed
Update test_parametrize_with_checks.py
1 parent 3fbd85e commit 765f36a

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

tests/integrations/sklearn/test_parametrize_with_checks.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,16 @@
44
from hyperactive.opt import GridSearch
55
from hyperactive.optimizers import RandomSearchOptimizer
66

7+
from sklearn.model_selection import KFold
78
from sklearn.utils.estimator_checks import parametrize_with_checks
89

9-
1010
svc = svm.SVC()
1111
parameters = {"kernel": ["linear", "rbf"], "C": [1, 10]}
1212
opt = RandomSearchOptimizer()
1313
hyperactivecv = HyperactiveSearchCV(svc, parameters, opt)
1414

15-
optcv = OptCV(estimator=svc, optimizer=GridSearch(param_grid=parameters))
15+
cv = KFold(n_splits=2, shuffle=True, random_state=42)
16+
optcv = OptCV(estimator=svc, optimizer=GridSearch(param_grid=parameters, cv=cv))
1617

1718
ESTIMATORS = [hyperactivecv, optcv]
1819

0 commit comments

Comments
 (0)