基于改进PSO的客流预测方法、介质及设备

本发明提出了一种基于改进PSO的客流预测方法、介质及设备，属于时间序列预测技术领域。本发明包括如下步骤：收集包含多个维度特征的数据并进行处理；通过注意力机制计算特征的注意力权重WATTENTION；通过粒子群优化算法计算特征的PSO权重WPSO；融合注意力权重WATTENTION和PSO权重WPSO，得到融合后的最终权重WFinal；通过最终权重WFinal对特征进行加权，然后输入到Transformer模型中，输出客流预测结果。本发明通过结合注意力机制和PSO进行特征优化，提高了预测性能；避免了手动挑选特征的主观性，增强了模型的适应性；充分考虑到适应数据特征的重要性随时间变化的情况；直观量化了特征对预测的贡献，帮助更好地理解预测结果。