一种基于Transformer的水库坝前垂向水温预测方法

本发明提供了一种基于Transformer的水库坝前垂向水温预测方法，包括以下步骤：步骤1，收集水库坝前垂向历史水温数据，并对历史水温数据进行格式转换获得格式化模型数据；步骤2，对格式化模型数据进行处理，然后搭建水温预测模型；步骤3，训练水温预测模型、测试水温预测模型；步骤4，测试集结果展示；步骤5，模型预测。本模型首先考虑了RevIN层将归一化数据变换至同一分布类型，可减小模型误差，然后采用Transformer模型，模型考虑了注意力机制，能够充分捕捉到不同时间段间水温分布差异，提高模型精度，最后是批次输出垂向水温数据，提高了预测效率。