老师我在运用lstm预测数据时,不知道这样做对不对

来源:10-4 LSTM网络基础

qq_慕姐45445

2021-07-25

  1. 列表项
  2. 列表项
  3. 列表项
  4. 列表项

这是一个LSTM模型预测股票,用5天的时间序列,3个特征的数据,预测下一天的值。我不是特别清楚在实际预测时怎么处理数据。能否帮忙看看。假如我用最后原数据的最后一组数据,5天3特征来预测,原数据打印如最后一张图片。

图中第一张图的1,2,3红框是我想使用原数据的最后1组数据来预测未来的y。
1框 截取原数据最后1组5*3 数据
2 框 数据tensor处理
3 框 数据reshape。这里开始不太有把握。torch.zeros(1,seq_len,3) 我自己想是1个batchsize,seq_len 5天序列,3是特征维度。这里用torch.zeros完全依据原来代码处理x,y 的方法。
然后就把这个处理后的R值直接给了Stock_Lstm模型了
原代码中还有数据处理的步骤Stock_dataset, Dataloader 。这里特别是 Dataloader是确定batchsize的。
我感觉不需要这2步了。直接把R喂给模型。
你看这有没有问题。完整代码需要展示吗

写回答

1回答

会写代码的好厨师

2021-07-27

要根据具体数据来确认模型是不是正确,不收敛或者过拟合,跟任务定义数据预处理关系很大,要针对具体问题具体分析。另外,label.的做法看着有问题,需要调整。网络结构上比较简单,可以做优化。其他看不出什么问题,可以qq私下沟通。

0
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程