Skip to content

twMisc/Covid-19-Machine-Learning-SIRD

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages