[发明专利]基于时空注意力图卷积网络的交通速度预测方法和装置

摘要

本发明公开的一种基于时空注意力图卷积网络的交通速度预测方法和装置，方法包括：对采集到现有速度数据集进行采样，获得有关时间序列的分量数据矩阵；构建时间注意力网络；将分量数据矩阵输入到时间注意力网络中，得到时间关联矩阵；构建空间注意力网络；将时间关联矩阵输入到空间注意力网络中，融合得到时空关联矩阵；将路网拓扑矩阵输入到图卷积网络，并与时空关联矩阵进行结合，进行模型训练；通过训练好的模型进行速度预测。通过将时间注意力网络与空间注意力网络融合形成时空注意力网络，提取交通网络在时间上跟空间上的关联性，同时结合图卷积网络不断融合邻接节点的信息，来对交通速度进行预测，提升对于交通速度预测的准确性。