0shelter0 / gman-pytorch

Implementation of Graph Muti-Attention Network with PyTorch

PyTorch implementation of GMAN: A Graph Multi-Attention Network for Traffic Prediction

This is a testing PyTorch version implementation of Graph Multi-Attention Network in the following paper: Chuanpan Zheng, Xiaoliang Fan, Cheng Wang, and Jianzhong Qi. "GMAN: A Graph Multi-Attention Network for Traffic Prediction", AAAI2020.

Requirements

• Python
• PyTorch
• Pandas
• Matplotlib
• Numpy

Dataset

The datasets could be unzipped and load from the data directory in this repository.

Citation

This version of implementation is only for learning purpose. For research, please refer to and cite from the following paper:

@inproceedings{ GMAN-AAAI2020,
  author = "Chuanpan Zheng and Xiaoliang Fan and Cheng Wang and Jianzhong Qi"
  title = "GMAN: A Graph Multi-Attention Network for Traffic Prediction",
  booktitle = "AAAI",
  pages = "1234--1241",
  year = "2020"
}