Graph

Implementation of Heterogeneous Graph Attention Network

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?

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