Skip to content

Commit

Permalink
Merge pull request optuna#4953 from not522/iris
Browse files Browse the repository at this point in the history
Replaces California housing dataset with iris dataset
  • Loading branch information
Alnusjaponica authored Sep 28, 2023
2 parents e06046b + 76b5837 commit ce84a78
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/integration_tests/lightgbm_tuner_tests/test_optimize.py
Original file line number Diff line number Diff line change
Expand Up @@ -727,9 +727,9 @@ def test_optuna_callback(self) -> None:
assert callback_mock.call_count == 10

def test_tune_best_score_reproducibility(self) -> None:
california = sklearn.datasets.fetch_california_housing()
iris = sklearn.datasets.load_iris()
X_trainval, X_test, y_trainval, y_test = train_test_split(
california.data, california.target, random_state=0
iris.data, iris.target, random_state=0
)

train = lgb.Dataset(X_trainval, y_trainval)
Expand Down Expand Up @@ -1058,9 +1058,9 @@ def test_get_best_booster_with_error(self) -> None:
tuner3.get_best_booster()

def test_tune_best_score_reproducibility(self) -> None:
california = sklearn.datasets.fetch_california_housing()
iris = sklearn.datasets.load_iris()
X_trainval, X_test, y_trainval, y_test = train_test_split(
california.data, california.target, random_state=0
iris.data, iris.target, random_state=0
)

train = lgb.Dataset(X_trainval, y_trainval)
Expand Down

0 comments on commit ce84a78

Please sign in to comment.