/pandemic_tgnn

Primary LanguagePythonMIT LicenseMIT

Data

Labels

We gather the ground truth for number of confiremed cases per region through open data for Italy, England, France and Spain. We have preprocessed the data and the final versions are in each country's subfolder in the data folder.

Graphs

The graphs are formed using the movement data from facebook Data For Good disease prevention maps. More specifically, we used the total number of people moving daily from one region to another, using the Movement between Administrative Regions datasets. We can not share the initial data due to the data license agreement, but after contacting the FB Data for Good team, we reached the consensus that we can share an aggregated and diminished version which was used for our experiments. These can be found inside the "graphs" folder of each country.These include the mobility maps between administrative regions that we use in our experiments until 12/5/2020, starting from 13/3 for England, 12/3 for Spain, 10/3 for France and 24/2 for Italy. The mapplots require the gadm1_nuts3_counties_sf_format.Rds file which can be found at the Social Connectedness Index data.

Code

Requirements

To run this code you will need the following python and R packages: numpy, pandas, scipy ,pytorch 1.5.1, pytorch-geometric 1.5.0, networkx 1.11, sklearn, dplyr, sf, ggplot2, sp.

Requirements for MAC

For MAC users, please use these versions: torch 1.7.0, torch-cluster 1.5.9 , torch-geometric 2.0.1 , torch-scatter 2.0.7, torch-sparse 0.6.12, torch-spline-conv 1.2.1., pystan 2.18.0.0 (for FB prophet).

Run

To run the experiments with the default settings:

cd code

python experiments.py
 
python metalearn.py
 

Use the script "gather_for_map.py" to aggregate data in the output folder to produce the map plots and the "tl_base.py" for the TL_BASE baseline. Use the "error_case_maps.R" to plot the maps of England (adjust it for the other countries).

Citation

If you find the methods or the datasets useful in your research, please consider adding the following citation:

@inproceedings{panagopoulos2020transfer,
  title={{Transfer Graph Neural Networks for Pandemic Forecasting}},
  author={Panagopoulos, George and Nikolentzos, Giannis and Vazirgiannis, Michalis},
  booktitle={Proceedings of the 35th AAAI Conference on Artificial Intelligence},
  year={2021},
}

License