Skip to content

Commit 08d41b2

Browse files
TEST: Backport adjustments to linear regression tolerances (#2245)
* TEST: Adjust tolerance for IncrementalLinearRegression fp32 intercept (#2199) * Adjust tolerance for IncrementalLinearRegression fp32 coefficients (#2204)
1 parent 5d4cf9a commit 08d41b2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onedal/linear_model/tests/test_incremental_linear_regression.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,10 @@ def test_full_results(queue, num_blocks, dtype):
9191
if queue and queue.sycl_device.is_gpu:
9292
tol = 5e-3 if model.coef_.dtype == np.float32 else 1e-5
9393
else:
94-
tol = 2e-3 if model.coef_.dtype == np.float32 else 1e-5
94+
tol = 3e-3 if model.coef_.dtype == np.float32 else 1e-5
9595
assert_allclose(coef, model.coef_.T, rtol=tol)
9696

97-
tol = 2e-3 if model.intercept_.dtype == np.float32 else 1e-5
97+
tol = 3e-3 if model.intercept_.dtype == np.float32 else 1e-5
9898
assert_allclose(intercept, model.intercept_, rtol=tol)
9999

100100
Xt = gen.random(size=(num_samples_test, num_features), dtype=dtype)

0 commit comments

Comments
 (0)