基于时空图注意力的交通流量预测方法及计算设备
摘要:
本发明公开了一种基于时空图注意力的交通流量预测方法,其特征在于,包括:S1.获取交通流数据集并进行预处理;S2.将经过预处理的数据输入预先训练好的交通流量预测模型得到隐藏特征,所述交通流量预测模型包含时间注意力模块和四个叠加时空卷积块ST‑Block;每个叠加时空卷积块ST‑Block包括两个门控扩张因果卷积模块,一个图注意力网络;S3.所述隐藏特征经过两组RelU函数和Linear完成隐藏层到输出层的线性变换以实现隐藏特征到交通流量预测结果的映射。本发明能够有效增强对交通路网中的时空相关性特征提取及分析,提高了预测的准确性和稳定性。
0/0