
STFGNN:STFGNN在AAAI-2021的代码实现(时空交通数据预测)。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
在AAAI-2021会议上,STFGNN的MXNet实现用于交通流量预测,该模型即为[用于交通流量预测的时空融合图神经网络,AAAI 2021]( )。 该框架的构建依赖于一个精心设计的管道,因此我们强烈建议您对其进行充分的熟悉。 为了便于使用,Python 3 参考了 requirements.txt 文件中的依赖项。 在几个公开可用的交通数据集上进行了验证,包括PEMS03、PEMS04、PEMS07和PEMS08。 为了获取所需数据,请使用密码p72z下载并解压缩 data.tar.gz 文件。 随后,针对PEMS03、PEMS04、PEMS07和PEMS08数据集进行模型训练,通过运行 python main_4n0_3layer_12T_res.py --config config/XXXX/individual_3layer_12T.json 命令来训练您自己的数据集。
全部评论 (0)
还没有任何评论哟~


