scratch_covid.py 366 Bytes
Newer Older
Chris Jewell's avatar
Chris Jewell committed
1
2
import numpy as np

Chris Jewell's avatar
Chris Jewell committed
3
4
from covid.rdata import *
from covid.model import CovidUKODE
Chris Jewell's avatar
Chris Jewell committed
5

Chris Jewell's avatar
Chris Jewell committed
6
7
8
9
K = load_age_mixing('data/polymod_normal_df.rds').to_numpy(dtype=np.float32)
T = load_mobility_matrix('data/movement.rds').to_numpy(dtype=np.float32)
np.fill_diagonal(T, 0.)
N = load_population('data/pop.rds')['n'].to_numpy(dtype=np.float32)
Chris Jewell's avatar
Chris Jewell committed
10

Chris Jewell's avatar
Chris Jewell committed
11
model = CovidUKODE(K, T, N, T.shape[0])
Chris Jewell's avatar
Chris Jewell committed
12