Skip to content

Commit

Permalink
Update test for LightGBM 4.x
Browse files Browse the repository at this point in the history
Signed-off-by: Beat Buesser <[email protected]>
  • Loading branch information
beat-buesser committed Dec 19, 2023
1 parent 8ff9a7d commit d9eb294
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/metrics/test_verification_decision_trees.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def test_LightGBM(self):
"verbose": 0,
}

model = lightgbm.train(parameters, train_data, valid_sets=test_data, num_boost_round=2, early_stopping_rounds=1)
model = lightgbm.train(parameters, train_data, valid_sets=test_data, num_boost_round=2)

classifier = LightGBMClassifier(model=model)

Expand Down

0 comments on commit d9eb294

Please sign in to comment.