Commit c939ddec authored by Chris Jewell's avatar Chris Jewell
Browse files

Merge branch 'master' into 'spatial-effect'

# Conflicts:
#   pyproject.toml
parents 9d10109c b72aeece
......@@ -109,14 +109,15 @@ def predict(
dims=["prediction_time"],
)
estimated_init_state, predicted_events = predicted_incidence(
samples,
initial_state,
covar_data,
initial_step,
num_steps,
out_of_sample,
)
with tf.device("CPU"): # TODO: work out effect GPU solution
estimated_init_state, predicted_events = predicted_incidence(
samples,
initial_state,
covar_data,
initial_step,
num_steps,
out_of_sample,
)
prediction = xarray.DataArray(
predicted_events.numpy(),
......
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