目录
一、基于PyTorch搭建LSTM模型实现风速时间序列预测
本专栏第二篇文章写的是利用LSTM构建模型进行时间序列预测,但是对于LSTM只能够从前向后推理,无法编码从后到前的信息,但是这一缺点被BiLSTM所弥补,通过BiLSTM可以更好的捕捉双向的信息依赖。
本篇则侧重于如何搭建双向LSTM,使用融合双向的信息进行建模从而达到风速预测。
二、配置类
下面是本项目需要使用的参数以及相关变量,为了方便我们将所有参数封装到一个类中,也可以使用 argparse
参数解析方式。
为了说明数据各个阶段的维度变化,特此定义了如下变量大小,小伙伴需要记住下面变量的值一遍理解下文说明各个阶段的维度大小。
class Config():
data_path = '../data/wind_dataset.csv'
timestep = 1 # 时间步长,就是利用多少时间窗口
batch_size = 32 # 批次大小
feature_size = 1 # 每个步长对应的特征数量,这里只使用1维,每天的风速
hidden_size = 256 # 隐层大小
outp