tests.test_issues.test_issue_29.test_issue_29()

Complexity

Conditions

1

Size

Total Lines	29
Code Lines	20

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

"""Test module for issue #29 reproduction."""
def test_issue_29():
    """Test issue 29 - accessing optimizer attributes during optimization."""
    data = load_diabetes()
    X, y = data.data, data.target

    def model(para):
        dtr = DecisionTreeRegressor(
            min_samples_split=para["min_samples_split"],
            max_depth=para["max_depth"],
        )
        scores = cross_val_score(dtr, X, y, cv=3)

        print(
            "Iteration:",
            para.optimizer.nth_iter,
            " Best score",
            para.optimizer.best_score,
        )

        return scores.mean()

    search_space = {
        "min_samples_split": list(range(2, 12)),
        "max_depth": list(range(2, 12)),
    }

    hyper = Hyperactive()
    hyper.add_search(model, search_space, n_iter=20)
    hyper.run()