added numerical stability to eta computation in ./models

Author: davidsebfischer

batchglm/models/base_glm/model.py

@@ -98,7 +98,9 @@ def eta_loc(self) -> np.ndarray:
 
     @property
     def eta_scale(self) -> np.ndarray:
-        return np.matmul(self.design_scale, self.b)
+        eta = np.matmul(self.design_scale, self.b)
+        eta = self.np_clip_param(eta, "eta_scale")
+        return eta
 
     @property
     def location(self):

batchglm/models/glm_nb/model.py

@@ -30,6 +30,7 @@ def eta_loc(self) -> np.ndarray:
         eta = np.matmul(self.design_loc, self.a)
         if self.size_factors is not None:
             eta += self.size_factors
+        eta = self.np_clip_param(eta, "eta_loc")
         return eta
 
     def eta_loc_j(self, j) -> np.ndarray:
@@ -39,6 +40,7 @@ def eta_loc_j(self, j) -> np.ndarray:
         eta = np.matmul(self.design_loc, self.a[:, j])
         if self.size_factors is not None:
             eta += self.size_factors
+        eta = self.np_clip_param(eta, "eta_loc")
         return eta
 
     # Re-parameterizations: