一种门控时空融合Transformer的交通流量预测方法、设备及存储介质

本发明公开了一种门控时空融合Transformer的交通流量预测方法、设备及存储介质，获取交通流量数据，将交通流量数据转化为包含时间维度、空间维度和交通流量特征维度的第一交通流量数据矩阵；将时间戳信息、周期信息、时间位置信息和空间位置信息融入到第一交通流量数据矩阵，得到第二交通流量数据矩阵；将第二交通流量数据矩阵输入到K个串行的时空编码模块，得到K个第三交通流量数据矩阵；将K个第三交通流量数据矩阵相加后输入卷积神经网络，得到交通流量预测结果。本发明提供的一种门控时空融合Transformer的交通流量预测方法、设备及存储介质，可以有效的提高交通流量预测的精度。