摘要
本发明公开了一种基于长短期Transformer模型的交通流量预测方法,用于交通路网中预测未来的交通情况。步骤如下:获取交通流量序列并对交通流量序列进行位置编码、时间编码,获取多角度特征表示;将交通流量序列的多角度特征表示输入长短期Transformer模型,所述长短期Transformer模型包括顺序连接的长短期注意力层、前馈网络层、残差连接层、归一化层、全连接层,得到输出的预测结果;根据预测结果和数据真实结果,以最小化损失函数为目标迭代训练,更新参数;用经过训练的长短期Transformer模型在测试集上得到交通流量预测结果。
- 单位