for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
"""Test module for sklearn parametrize_with_checks integration."""
from sklearn import svm
from sklearn.model_selection import KFold
from sklearn.utils.estimator_checks import parametrize_with_checks
from hyperactive.integrations import HyperactiveSearchCV, OptCV
from hyperactive.opt import GridSearchSk as GridSearch
from hyperactive.optimizers import RandomSearchOptimizer
svc = svm.SVC()
parameters = {"kernel": ["linear", "rbf"], "C": [1, 10]}
opt = RandomSearchOptimizer()
hyperactivecv = HyperactiveSearchCV(svc, parameters, opt)
cv = KFold(n_splits=2, shuffle=True, random_state=42)
optcv = OptCV(estimator=svc, optimizer=GridSearch(param_grid=parameters), cv=cv)
ESTIMATORS = [hyperactivecv, optcv]
@parametrize_with_checks(ESTIMATORS)
def test_estimators(estimator, check):
"""Test estimators with sklearn estimator checks."""
check(estimator)