diff --git a/dlordinal/layers/clm.py b/dlordinal/layers/clm.py index acbe4f5..e7537c1 100644 --- 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)