Added spatial effect to R calculation

chrism0dwk · May 5, 2021 · 5ded6b1 · 5ded6b1
1 parent 5bf6108
commit 5ded6b1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/covid19uk/model_spec.py b/covid19uk/model_spec.py
@@ -343,7 +343,11 @@ def fn(t, state):
             ),
         )
 
-        eta = alpha_t_ + param["beta_area"] * log_area[:, tf.newaxis]
+        eta = (
+            alpha_t_
+            + param["beta_area"] * log_area[:, tf.newaxis]
+            + param["sigma_space"] * param["spatial_effect"]
+        )
         infec_rate = (
             tf.math.exp(eta)
             * (