基于联邦时空图学习的交通流量预测及模型训练方法

本公开的实施例公开了基于联邦时空图学习的交通流量预测及模型训练方法。该方法的具体实施方式包括：基于本地客户端存储的历史交通流量生成时空图的节点特征矩阵；基于节点特征矩阵、预先训练的节点嵌入矩阵和预先训练的时间嵌入矩阵确定时空依赖关系；将时空依赖关系发送到服务器，并接收服务器返回的聚合时空依赖关系，其中，聚合时空依赖关系是由服务器将来自不同客户端的时空依赖关系进行聚合得到的；基于聚合时空依赖关系通过预先训练的预测模型确定未来的交通流量。该实施方式能够有效地捕捉客户端之间和客户端内部的时空依赖关系，以提高交通流量预测的准确性。