一种基于时空图卷积网络的交通流量预测方法

    公开(公告)号:CN115293399A

    公开(公告)日:2022-11-04

    申请号:CN202210684371.0

    申请日:2022-06-16

    摘要: 本发明公开了一种基于时空图卷积网络的交通流量预测方法,包括S1):利用道路传感器收集目标区域的交通流量数据并进行预处理;S2):收集额外信息并进行编码,所述额外信息包括气候数据、当天是否为工作日以及节点周围的公共场所的种类和数量;S3):根据步骤S1)得到的预处理后的交通流量数据生成时间序列数据;S4):构建两个自适应向量ES,ET作为节点的空间嵌入表示,生成空间相关性矩阵;S5):构建基于DSGRU单元的DSRNN网络模型,将步骤S3)得到的时间序列数据作为输入和标签,再将步骤S2)得到的额外信息向量输入到模型中,辅助模型对目标区域的交通流量进行预测,可更好捕捉交通流量数据的动态空间依赖,并达到更好的预测效果。