时间序列lstm模型python代码
时间: 2024-11-23 12:29:53 浏览: 10
时间序列LSTM模型是一种广泛应用于时间序列预测的神经网络模型。下面是一个简单的Python代码示例,演示如何使用Keras库实现时间序列LSTM模型:
```python
from keras.models import Sequential
from keras.layers import LSTM, Dense
# 定义LSTM模型
model = Sequential()
model.add(LSTM(50, input_shape=(X_train.shape, X_train.shape)))
model.add(Dense(1))
model.compile(loss='mae', optimizer='adam')
# 训练模型
history = model.fit(X_train, y_train, epochs=50, batch_size=72, validation_data=(X_test, y_test), verbose=2, shuffle=False)
# 使用模型进行预测
yhat = model.predict(X_test)
```
在这段代码中,我们使用Keras库中的Sequential模型来定义时间序列LSTM模型。首先,我们添加一个LSTM层,其中包含50个LSTM单元。接下来,我们添加一个全连接层来输出模型的预测结果。最后,我们使用mean absolute error(MAE)作为损失函数,并使用Adam优化器进行优化。
在训练过程中,我们使用fit()函数来训练模型,并将训练集和测试集传递给函数。在每个epoch结束后,我们可以记录模型的训练和验证误差。
在模型训练完成后,我们可以使用predict()函数来进行预测,并将结果存储在yhat变量中。
阅读全文