Change/rewrite at.nnet functions

Priesemann-Group · Mar 1, 2023 · 9bffeba · 9bffeba
1 parent 7649efd
commit 9bffeba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/covid19_inference/model/utility.py b/covid19_inference/model/utility.py
@@ -143,7 +143,7 @@ def tt_lognormal(x, mu, sigma):
     """
     Calculates a lognormal pdf for integer spaced x input.
     """
-    x = at.nnet.relu(x - 1e-12) + 1e-12  # clip values at 1e-12
+    x = at.maximum(x, 1e-12)  # clip values at 1e-12
     distr = 1 / x * at.exp(-((at.log(x) - mu) ** 2) / (2 * sigma**2))
 
     # normalize, add a small offset in case the sum is zero
@@ -159,7 +159,7 @@ def tt_gamma(x, mu=None, sigma=None, alpha=None, beta=None):
     if alpha is None and beta is None:
         alpha = mu**2 / (sigma**2 + 1e-8)
         beta = mu / (sigma**2 + 1e-8)
-    x = at.nnet.relu(x - 1e-12) + 1e-12  # clip values at 1e-12
+    x = at.maximum(x, 1e-12)  # clip values at 1e-12
     distr = beta**alpha * x ** (alpha - 1) * at.exp(-beta * x)
 
     # normalize, add a small offset in case the sum is zero

diff --git a/covid19_inference/model/week_modulation.py b/covid19_inference/model/week_modulation.py
@@ -338,7 +338,7 @@ def by_weekday_modulation(cases, model=None):
     )
     r_week = r_base + delta_r
 
-    r_transformed_week = at.nnet.sigm.sigmoid(r_week)
+    r_transformed_week = at.math.sigmoid(r_week)
     pm.Deterministic("fraction_delayed_by_weekday", r_week)
 
     t = np.arange(model.sim_shape[0]) + model.sim_begin.weekday()  # Monday @ zero