Update tau max.

rapidsai · Oct 8, 2023 · 1ff4779 · 1ff4779
1 parent 439a83c
commit 1ff4779
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/legateboost/models/linear.py b/legateboost/models/linear.py
@@ -67,7 +67,7 @@ def solve_singular(self, a, b):
                 return res
             except (np.linalg.LinAlgError, cn.linalg.LinAlgError):
                 tau = max(tau * 2, eps)
-            if tau > 100:
+            if tau > 1e10:
                 raise ValueError(
                     "Numerical instability in linear model solve. "
                     "Consider normalising your data."