- Optionally replace
data/data.csv
with your own dataset of the same format. The original data for Russia was obtained here. - Run the data preparation notebook
- Run seir_moscow_hidden.ipynb to train and evaluate the usual SARS-COV-2 model.
- Run seir_two_strain_hidden_states_moscow.ipynb to use that model to forecast for two strains.
Articles: