Autofomer预测值反归一化

2023-02-25 07:15| 来源: 网络整理| 查看: 265

Autofomer源码中默认输出的预测值为标准化的数据，但我们往往需要输出实际的真实值来使用。

只需要在源码exp-main.py中加上以下代码即可：

在函数def _get_data(self, flag): data_set, data_loader = data_provider(self.args, flag) #return data_set, data_loader self.Data=data_set return data_set, data_loader 加上代码： self.Data=data_set

在vail，test方法中分别加入以下代码：

mean_X, std_X = self.Data.scaler.mean_,self.Data.scaler.scale_ pred = outputs.detach().cpu()*std_X+mean_X true = batch_y.detach().cpu()*std_X+mean_X mean_X, std_X = self.Data.scaler.mean_, self.Data.scaler.scale_ outputs = outputs * std_X + mean_X batch_y = batch_y * std_X + mean_X

然后就可以输出真实值了: 在这里插入图片描述

【本文地址】

公司简介

联系我们