上传者: 42120997
|
上传时间: 2022-04-20 10:37:53
|
文件大小: 22.4MB
|
文件类型: ZIP
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