立即打开
STFGNN的MXNet实现时空融合图神经网络的交通流量预测公共交通数据集
要求这是STFGNN的MXNet实现:[用于交通流量预测的时空融合图神经网络,AAAI 2021]( )。 该框架基于框架构建。 强烈建议您熟悉其管道。Python3参见requirements.txt资料准备STFGNN在这几个公共交通数据集上实现。来自PEMS03 , PEMS04 , PEMS07和PEMS08 。 使用密码p72z下载数据 ,并使用tar -zxvf data.tar.gz p72z tar -zxvf data.tar.gz解压缩数据文件。模型训练PEMS03,PEMS04,PEMS07,PEMS08:python main_4n0_3layer_12T_res.py --config config/XXXX/individual_3layer_12T.json训练自己的数据集
扫码加我 拉你入群
请注明:姓名-公司-职位
以便审核进群资格,未注明则拒绝
栏目导航
热门文章
推荐文章
扫码加好友,拉您进群