Commit 7b44b542 authored by Chris Jewell's avatar Chris Jewell
Browse files

Reduce spatial correlation

parent 88ebe233
......@@ -171,7 +171,7 @@ def CovidUK(covariates, initial_state, initial_step, num_steps):
def spatial_effect():
W = tf.convert_to_tensor(covariates["adjacency"])
Dw = tf.linalg.diag(tf.reduce_sum(W, axis=-1)) # row sums
rho = 0.5
rho = 0.25
precision = Dw - rho * W
cov = tf.linalg.inv(precision)
scale = tf.linalg.cholesky(cov)
......
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