Commit 5ded6b11 authored by Chris Jewell's avatar Chris Jewell
Browse files

Added spatial effect to R calculation

parent 5bf61083
......@@ -343,7 +343,11 @@ def next_generation_matrix_fn(covar_data, param):
),
)
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)
* (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment