基于时空特征嵌入与门操作优化的交通预测方法及装置

本发明属于交通数据预测技术领域，具体涉及一种基于时空特征嵌入与门操作优化的交通预测方法及装置。所述方法包括：引入时间步特征关系嵌入和时间步空间关系嵌入，生成一个同时考虑时间和空间的综合特征表示，再结合历史交通节点数据，得到融合特征；构建交通数据预测模型，所述交通数据预测模型包括图卷积网络和门控循环单元；其中，所述图卷积网络能够捕捉节点之间的空间关系，所述门控循环单元能够捕捉时间上的依赖关系；基于融合特征对交通数据预测模型进行训练，采用训练好的交通数据预测模型，预测出未来交通数据。本发明能够更好地捕捉复杂交通网络中的时空关系，从而提供更准确和可靠的交通预测。