Using Graph neural network to predict spread of COVID-19 from partial tests

Relavent Codes that I am working on currently:

  • gnn_seird_cluster.py : Contains the GNN model and Recurrent GNN model on static graphs.
  • rgnn_seird.py: Contains the sequence modelling on graphs and associated data processing.