GMAN 的 PyTorch 实现:用于交通预测的图多注意网络
这是以下论文中图多注意力网络的测试 PyTorch 版本实现:Chuanpan Zheng、Xiaoliang Fan、Cheng Wang 和 Jianzhong Qi。 “ ”,AAAI2020。
要求
Python
火炬
熊猫
Matplotlib
麻木
数据集
数据集可以解压缩并从此存储库中的数据目录加载。
引文
此版本的实现仅用于学习目的。 有关研究,请参考并引用以下论文:
@inproceedings{ GMAN-AAAI2020,
author = "Chuanpan Zheng and Xiaoliang Fan and Cheng Wang and Jianzhong Qi"
title = "GMAN: A Graph Multi-Attention Network for Traffic
2022-04-20 10:37:53
22.4MB
Python
1