一种基于掩码预训练与提示调优的跨城市小样本时空图预测方法

本发明提出一种基于掩码预训练与提示调优的跨城市小样本时空图预测方法，包括以下步骤：S1：在多个数据丰富的源城市上进行时空掩码预训练，获得预训练的空间编码器和时间编码器；S2：以所述预训练的空间编码器和时间编码器为基础，结合提示网络、图结构生成器以及下游时空图神经网络，构建跨城市小样本时空图预测模型；S3：利用源城市数据对所述跨城市小样本时空图预测模型进行元训练；S4：利用目标城市数据对元训练后的跨城市小样本时空图预测模型进行微调；将输入数据馈送至元训练后的跨城市小样本时空图预测模型中以获得预测结果，计算梯度并更新模型参数，直至模型达到收敛状态。本发明可以更好地实现小样本时空图预测。