From d285d6ba2a4912f105a87405e831be13d37033fa Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Mon, 20 Jun 2022 23:55:12 +0800 Subject: [PATCH] Reduce regularization in GPU gblinear test. (#8010) --- tests/python-gpu/test_gpu_linear.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/python-gpu/test_gpu_linear.py b/tests/python-gpu/test_gpu_linear.py index af8fe1bbe..7bad8b72f 100644 --- a/tests/python-gpu/test_gpu_linear.py +++ b/tests/python-gpu/test_gpu_linear.py @@ -42,9 +42,13 @@ class TestGPULinear: # Loss is not guaranteed to always decrease because of regularisation parameters # 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, 1.0), - strategies.floats(1e-5, 1.0)) + @given( + parameter_strategy, + strategies.integers(10, 50), + tm.dataset_strategy, + strategies.floats(1e-5, 0.8), + strategies.floats(1e-5, 0.8) + ) @settings(deadline=None, print_blob=True) def test_gpu_coordinate_regularised(self, param, num_rounds, dataset, alpha, lambd): assume(len(dataset.y) > 0)