基于双阶段堆叠图卷积网络的交通流预测模型构建方法

摘要

本发明公开了基于双阶段堆叠图卷积网络的交通流预测模型构建方法，包括以下步骤：首先对原始交通数据进行预处理，然后建立基于双阶段堆叠图卷积网络的交通流预测模型构建方法EDGCN，编码器和解码器两个阶段都使用时空图卷积网络EDGCN，时空图卷积网络EDGCN里面包括一个门控递归单元GRU和两个AGCN，而AGCN主要由图卷积网络GCN、门控线性单元GTU及注意力机制构成；最后，使用预处理好的数据训练所建立的交通流多步预测模型，并调整模型中的参数以获得最优的交通流多步预测效果；实验结果验证了模型在交通流多步预测任务上的有效性，并为使用深度学习解决多步预测任务提供新的思路和方法。