gnn-covid-vaccine

Modified STAN model (https://github.com/v1xerunt/STAN) on the country scale with SIRVC (SIR + vaccinations + community interactions) loss.

International interactions modeled using crowdsourced flight data (https://zenodo.org/record/4737390#.YMuJUGhKiF4).

Name		Name	Last commit message	Last commit date
Latest commit History 25 Commits
.vscode		.vscode
__pycache__		__pycache__
all_models		all_models
plots		plots
processed_flights		processed_flights
results		results
save		save
stan_results		stan_results
Countries-Continents.csv		Countries-Continents.csv
README.md		README.md
ensemble_country_to_world.ipynb		ensemble_country_to_world.ipynb
ensemble_train_continents.ipynb		ensemble_train_continents.ipynb
ensemble_train_world.ipynb		ensemble_train_world.ipynb
model.py		model.py
model_add_world.py		model_add_world.py
model_stan.py		model_stan.py
neighbors.csv		neighbors.csv
plot.ipynb		plot.ipynb
plot_decomposition.ipynb		plot_decomposition.ipynb
random.ipynb		random.ipynb
stan_sirv.ipynb		stan_sirv.ipynb
stan_sirvc.ipynb		stan_sirvc.ipynb
stan_sirvc_country_layer.py		stan_sirvc_country_layer.py
stan_sirvc_layer.py		stan_sirvc_layer.py
stan_sirvc_train_sequential.ipynb		stan_sirvc_train_sequential.ipynb
stan_sirvc_world_input.ipynb		stan_sirvc_world_input.ipynb
stan_sirvc_world_predict.ipynb		stan_sirvc_world_predict.ipynb
stan_tree_decomp.ipynb		stan_tree_decomp.ipynb

Provide feedback