tests.integrations.sklearn.test_parametrize_with_checks

Complexity

Total Complexity

1

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	test_estimators()	0	4	1

"""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)