Skip to content

Commit

Permalink
Adjust tolerance for IncrementalLinearRegression fp32 coefficients (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
david-cortes-intel authored Dec 5, 2024
1 parent 48b70a1 commit 0fe0384
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def test_full_results(queue, num_blocks, dtype):
if queue and queue.sycl_device.is_gpu:
tol = 5e-3 if model.coef_.dtype == np.float32 else 1e-5
else:
tol = 2e-3 if model.coef_.dtype == np.float32 else 1e-5
tol = 3e-3 if model.coef_.dtype == np.float32 else 1e-5
assert_allclose(coef, model.coef_.T, rtol=tol)

tol = 3e-3 if model.intercept_.dtype == np.float32 else 1e-5
Expand Down

0 comments on commit 0fe0384

Please sign in to comment.