HetGAN

Implementation of Heterogeneous Graph Attention Network

This is the code repository of paper "Prediction of Metro Ridership During the COVID-19 Pandemic: A Deep Learning Approach Based on Heterogeneous Graph" which is submitted to Transportation Rsearch Part C: Emgerging Technologies.

Datasets

• New York City COVID case number
• New York City Metro Ridership Number

Model Framework

How to Run?

• Single run:

cd Experiments/
python main.py --epoch 200 --seed 1230 --log_file='./logs/running_log.log'

• Grid Search:

  cd Experiments/
  python grid_search.py --epoch 200 --seed 1230 --log_file='./logs/grid_search.log'