基于图卷积及时空交错注意力机制的水质预测方法

本发明适用于水质预测领域，具体提供了基于图卷积及时空交错注意力机制的水质预测方法，该方法中，基于时空交错注意力机制构建包含四层Transformer架构的预测模型；第一层为嵌入Embedding；第二层为位置编码模块；第三层为N个相同带有时空交错注意力机制的Transformer编码器模块，在时空交错注意力机制中，通过将图卷积融入Transformer框架，将时间注意力模块与空间注意力模块进行堆叠，构成Transformer编码器模块；第四层为输出层；对构建好的预测模型进行训练，训练好的模型用于对目标水质进行预测。本发明可同时处理数据的时间和空间维度，捕捉不同时空尺度下的依赖关系，提高模型的表达能力和预测性能。