[发明专利]一种城市流量预测方法

摘要

本发明涉及智慧城市领域，特别公开了一种城市流量预测方法。该城市流量预测方法，具体包括如下步骤：对城市流量数据按时间顺序以窗口滑动方式进行划分，并构造时空随机遮掩数据和序列打乱数据；利用时空随机遮掩编码器EncM进行时空随机遮掩预训练；利用全局与局部相结合的注意力模块EncC进行序列顺序对比学习预训练；将时空随机遮掩预训练中自注意力模块产生的注意力权重处理为城市区域间相关性系数，利用图处理模块增强空间相关性捕捉能力；将预训练模块与图处理模块融合，构建时空融合预测模块，进行预测训练。本发明为两种预训练模块设计了合适的解码器，通过对不同角度捕获的时空特征进行解码，从而对下一时刻的城市流量进行预测。