Beispiel #1
0
    def test_set_params(self, teardown):
        step = LogisticRegression()

        new_params_wrong = {"non_existent_param": 42}
        with pytest.raises(ValueError):
            step.set_params(**new_params_wrong)

        new_params = {"C": 100.0, "fit_intercept": False, "penalty": "l1"}

        step.set_params(**new_params)
        params = step.get_params()

        expected = {
            "C": 100.0,
            "class_weight": None,
            "dual": False,
            "fit_intercept": False,
            "intercept_scaling": 1,
            "max_iter": 100,
            "multi_class": "warn",
            "n_jobs": None,
            "penalty": "l1",
            "random_state": None,
            "solver": "warn",
            "tol": 0.0001,
            "verbose": 0,
            "warm_start": False,
            "l1_ratio": None,
        }

        assert expected == params
Beispiel #2
0
    def test_get_params(self, teardown):
        step = LogisticRegression()
        params = step.get_params()

        expected = {
            "C": 1.0,
            "class_weight": None,
            "dual": False,
            "fit_intercept": True,
            "intercept_scaling": 1,
            "max_iter": 100,
            "multi_class": "warn",
            "n_jobs": None,
            "penalty": "l2",
            "random_state": None,
            "solver": "warn",
            "tol": 0.0001,
            "verbose": 0,
            "warm_start": False,
            "l1_ratio": None,
        }

        assert expected == params