Skip to content

Latest commit

 

History

History
23 lines (16 loc) · 797 Bytes

README.md

File metadata and controls

23 lines (16 loc) · 797 Bytes

Covid-19-Machine-Learning-SIRD

Reproducing the model proposed from https://arxiv.org/abs/2004.09478 with python and tensorflow.

Dataset is downloaded from https://github.com/CSSEGISandData/COVID-19/.

How to run

Run train_model.py to train a model.

Run draw_figure.py to outout figures with a trained model.

Run train_all.py as an example to train many countries(US, Spain, Belgium, China, France, Germany, United Kingdom, Italy)

Run draw_all.py as an example to draw figures for many models.

Run compare_model.py to output a Plotly compare of parameters

Folders

models

./models contains some pretrained model

html

./html contains some comparing results of models for different countries using Plotly, download and open with web browser

pdf

./pdf contains a presentation file