Autoformer pytorch代码解读(3)dataloader部分 transformer学习笔记

  • Post author:
  • Post category:其他


到这里产生一个疑问,为什么传入的x_dec只用到了他的维度,并没有用到这个数据本身呢?

从train开始进入,看看数据是怎么样的。


(1)进入了exp_main中的def train中。

(2)进入了exp_main中的这个get data中。

这个get data这个函数,调用的是data provider这个方法,他需要的参数是args和,flag。flag代表着是训练还是测试还是验证。

(3)进入到了这个datafactory这个里面

(4) self.seq_len = 96   self.label_len = 48    self.pred_len = 96

(5)到了data_loader的read_data这里。

这里最先获得的df_raw是(17420,8)维度的。

border1s[0,8544,11424],border2s[8640,11520,1440]。

broder1最后是0,broder2是8640。

到了这一步,这个df_data就是这个原始的数据。只不过是从第1列开始,也就是说没有第0列。

这里是得到标准化后的训练数据和测试数据。

为什么要这样呢?

对应的就是下面的三行。

接下来是对时间进行时间戳编码

data_stamp这个的维度是(8640,4),如下图所示。

这个data_x和这个data_y的维度都是(8640,7)的维度的数据。这个data_stamp的维度是(8640,4)的维度。



版权声明:本文为weixin_45193103原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。