# GMAN

**Repository Path**: verigle/GMAN

## Basic Information

- **Project Name**: GMAN
- **Description**: GMAN: A Graph Multi-Attention Network for Traffic Prediction
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 0
- **Created**: 2020-07-26
- **Last Updated**: 2026-03-30

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# GMAN: A Graph Multi-Attention Network for Traffic Prediction

<p align="center">
  <img width="600" height="450" src=./figure/GMAN.png>
</p>

This is the 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](https://ojs.aaai.org/index.php/AAAI/article/view/5477)", AAAI2020

## Data
The datasets are available at [Google Drive](https://drive.google.com/open?id=10FOTa6HXPqX8Pf5WRoRwcFnW9BrNZEIX) or [Baidu Yun](https://pan.baidu.com/s/14Yy9isAIZYdU__OYEQGa_g), provided by [DCRNN](https://github.com/liyaguang/DCRNN), and should be put into the corresponding `data/` folder.

## Requirements
Python 3.7.10, tensorflow 1.14.0, numpy 1.16.4, pandas 0.24.2

## Results
<p align="center">
  <img width="900" height="400" src=./figure/results.png>
</p>

## Third-party re-implementations
A Pytorch implementaion by [VincLee8188](https://github.com/VincLee8188) is available at [GMAN-Pytorch](https://github.com/VincLee8188/GMAN-PyTorch).

## Citation

If you find this repository useful in your research, please cite 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"
}
```