基于注意力机制的时空图卷积网络客流量预测方法

本发明公开了一种基于注意力机制的时空图卷积网络客流量预测方法，包括：客流量数据预处理输入、利用时间注意力机制和空间注意力机制捕捉客流量序列的依赖性、构建时空卷积块、时空图模型搭建、模型训练和客流量预测；其有益效果为：通过将时间注意机制与时空卷积结合，获取站点间空间相关程度和客流量信息的时序关联性，时空卷积块融合了空间维度的图卷积和时间维度的标准卷积，能够有效地捕捉客流量数据的动态时空特征，提高了预测精度，采用的基于注意力机制的时空图卷积网络模型完全由卷积结构组成，以能够更少的参数和更快的训练速度实现了对输入的并行化。