三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

从LSTM到Informer的长序列预测实战

从LSTM到Informer的长序列预测实战

摘要

“给我10年的股票日线,预测未来5天的开盘、收盘、最高、最低涨跌幅”——看似简单的时间序列任务,在实际落地中却充满挑战:单一模型难以捕捉长程依赖,固定训练集浪费大量后期数据,点预测无法量化不确定性。本文从基础的LSTM出发,逐步演进到Transformer、PatchTST,最后在充足算力下选择了Informer,并配合Walk‑Forward 滚动训练让整整10年的数据全部发挥作用。文章提供完整的数据构造、特征工程、Informer概率建模以及Walk‑Forward回测代码,可直接落地。


一、问题定义与数据构造

1.1 预测目标

我们要利用过去一段连续交易日的四种涨跌幅(相对于前一日收盘)来预测未来5日的这四种涨跌幅。

1.2 计算四种涨跌幅

close=df['close'].values ret_close
← 返回列表