Autofomer预测值反归一化

最新推荐文章于 2024-06-11 16:39:22 发布

凯天

最新推荐文章于 2024-06-11 16:39:22 发布

阅读量841

点赞数 6

文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/weixin_43149598/article/details/129109974

版权

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

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