Fixed CLM numerical instability

ayrna · Jun 6, 2024 · 7fc0908 · 7fc0908
1 parent 1527b2b
commit 7fc0908
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlordinal/layers/clm.py b/dlordinal/layers/clm.py
@@ -91,7 +91,7 @@ def _clm(self, projected: torch.Tensor, thresholds: torch.Tensor):
             0,
             1,
         )
-        z3 = a - b
+        z3 = torch.clip(a - b, -10, 10)
 
         if self.link_function == "probit":
             a3T = self.dist.cdf(z3)