一种基于Transformer的客流预测方法与装置

本申请公开了一种基于Transformer的客流预测方法与装置，涉及交通流预测技术领域。该方法包括：获取历史客流数据以及外部信息数据；对外部信息数据进行编码得到外部信息向量，对历史客流数据进行分片处理得到分片数据，对分片数据进行线性投影后与外部信息数据融合得到融合特征；为Transformer编码层构建位置编码器以使Transformer模型理解输入数据的时间顺序，并基于Transformer模型提取融合特征的深层特征；利用二维卷积层对深层特征进行解码，输出客流预测结果。该方法实现了在面对多种人为影响因素和自然不可抗力因素导致的突发交通状况时，可以及时、准确地做出客流预测判断。