Skip to content

Commit

Permalink
Avoid over regularization.
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Jan 17, 2022
1 parent 45bd7d0 commit 4f5eefb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/python-gpu/test_gpu_linear.py
Original file line number Diff line number Diff line change
@@ -43,8 +43,8 @@ def test_gpu_coordinate(self, param, num_rounds, dataset):
# We test a weaker condition that the loss has not increased between the first and last
# iteration
@given(parameter_strategy, strategies.integers(10, 50),
tm.dataset_strategy, strategies.floats(1e-5, 2.0),
strategies.floats(1e-5, 2.0))
tm.dataset_strategy, strategies.floats(1e-5, 1.0),
strategies.floats(1e-5, 1.0))
@settings(deadline=None)
def test_gpu_coordinate_regularised(self, param, num_rounds, dataset, alpha, lambd):
assume(len(dataset.y) > 0)

0 comments on commit 4f5eefb

Please sign in to comment.