《TF-GNN:Graph Neural Networks》附录《A.2.2 Creating GraphTensors》例程
2022-10-09 20:05:19 2KB TFGNN
1
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