一种基于时空图神经网络的车流预测方法及相关设备

本申请公开了一种基于时空图神经网络的车流预测方法及相关设备，涉及人工智能技术领域，方法包括：获取交通路网图；其中，交通路网图的节点表示各个交通站点，节点之间的边表示对应交通站点之间的道路；动态获取各个交通站点的时间序列数据并对时间序列数据进行归一化预处理；其中，时间序列数据包括对应交通站点的车流量和车速；利用归一化预处理后的时间序列数据动态更新交通路网图，得到动态更新图；利用图注意力网络提取得到动态更新图的空间特征；利用三个时间尺度不相同的长短时记忆网络对空间特征提取对应尺度的时间特征；根据三个尺度的时间特征预测得到下一个时间步的车流量。本申请能更准确预测车流量。