公开(公告)号：CN119048317A

公开(公告)日：2024-11-29

申请号：CN202411199596.2

申请日：2024-08-29

Applicant: 昆明理工大学

Inventor： 杨芳 ,  帅春燕 ,  杨扬 ,  张小七 ,  张婷 ,  殷奇 ,  段宋 ,  赵龙周

IPC: G06Q50/40 ,  G06F18/25 ,  G06N3/042 ,  G06N3/0442 ,  G06F18/213 ,  G06N3/045 ,  G06N3/08

Abstract: 本发明实施例公开了一种基于时空图注意力神经网络的城市轨道交通OD预测方法，包括：基于城市轨道交通网络的空间特征数据集和时序特征数据集对时空图注意力神经网络模型进行模型训练；将城市轨道交通网络的客流OD时序数据作为原始OD序列，输入训练好的时空图注意力神经网络模型中基于多头注意力机制进行特征提取，得到客流OD时序数据的空间特征OD序列和时间特征OD序列；使用双向门控循环单元模型对原始OD序列、空间特征OD序列和时间特征OD序列进行融合并预测得到城市轨道交通网络的客流OD预测结果。可以在城市轨道交通中，提升客流OD序列预测性能。