AAAI-2021时空融合图神经网络的交通流量预测
要求
这是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
训练自己的数据集
(
2021-10-17 09:38:58
26.59MB
Python
1