简介:GMAN-PyTorch是一款使用PyTorch框架构建的高效图注意力网络工具包,专门针对大规模图数据设计,提供强大的节点分类、链接预测等机器学习功能。
GMAN的PyTorch实现:用于交通预测的图多注意网络
这是以下论文中的图多注意力网络(Graph Multi-Attention Network, GMAN)的一个测试版PyTorch实现:
Chuanpan Zheng、Xiaoliang Fan、Cheng Wang 和 Jianzhong Qi。 GMAN: A Graph Multi-Attention Network for Traffic,AAAI2020。
要求:Python 火炬(torch)、熊猫(pandas)、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,
}