Python DeadZoneRegressor Examples

Example #1

import pytest
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler

from sklego.dummy import RandomRegressor
from sklego.linear_model import DeadZoneRegressor
from sklego.mixture import GMMClassifier, BayesianGMMClassifier, GMMOutlierDetector, BayesianGMMOutlierDetector
from tests.conftest import id_func


@pytest.mark.parametrize("estimator", [
    RandomRegressor(strategy="uniform"),
    RandomRegressor(strategy="normal"),
    DeadZoneRegressor(effect="linear", n_iter=100),
    DeadZoneRegressor(effect="quadratic", n_iter=100),
],
                         ids=id_func)
def test_shape_regression(estimator, random_xy_dataset_regr):
    X, y = random_xy_dataset_regr
    assert estimator.fit(X, y).predict(X).shape[0] == y.shape[0]
    pipe = Pipeline(steps=[('scaler', StandardScaler()), ('clf', estimator)])
    assert pipe.fit(X, y).predict(X).shape[0] == y.shape[0]


@pytest.mark.parametrize("estimator", [
    GMMClassifier(),
    BayesianGMMClassifier(),
    GMMOutlierDetector(threshold=0.999, method="quantile"),
    GMMOutlierDetector(threshold=2, method="stddev"),
    BayesianGMMOutlierDetector(threshold=0.999, method="quantile"),
    BayesianGMMOutlierDetector(threshold=2, method="stddev")

Example #2

File: test_deadzone.py Project: prateekchandrajha/scikit-lego

def mod(request):
    return DeadZoneRegressor(effect=request.param, n_iter=1000)

Example #3

File: test_deadzone.py Project: prateekchandrajha/scikit-lego

def test_deadzone(test_fn):
    regr = DeadZoneRegressor(n_iter=10)
    test_fn(DeadZoneRegressor.__name__, regr)

Example #4

File: test_deadzone.py Project: tsuhina/scikit-lego

def mod(request):
    return DeadZoneRegressor(effect=request.param)