[发明专利]基于动态时空图卷积注意力模型的交通预测方法及装置

摘要

本发明公开了一种基于动态时空图卷积注意力模型的交通预测方法及装置，首先，将道路交通网络建模为图结构，利用图结构和图信号对动态时空图卷积注意力模型进行建模，通过所述动态时空图卷积注意力模型的编码器单元，捕获不同时刻传感器节点间的动态空间相关性，捕获历史时刻步长的时间相关性以及多步预测间的时间相关性，以及得到包含每个预测步长的隐藏状态；解码器单元接收每个预测步长的隐藏状态，采用调度采样策略，得到未来一小时所有传感器节点记录速度值的预测结果。实验证明，本发明方法能有效得到未来一小时的交通预测结果，本发明方法所述模型超越了SOTA。