举一个Python 实现LSTM的例子
时间: 2023-11-19 18:45:58 浏览: 74
lstm 成功,lstm实现,Python
5星 · 资源好评率100%
可以参考以下代码,实现一个简单的LSTM模型:
```python
import keras
from keras.models import Sequential
from keras.layers import Dense, LSTM
# 构建模型
model = Sequential()
model.add(LSTM(32, input_shape=(10, 2))) # 10是序列长度,2是输入特征维度
model.add(Dense(1, activation='sigmoid'))
# 编译模型
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['accuracy'])
# 训练模型
X = [[[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]] # 一个样本序列,其中每个序列有5个时间步,每个时间步有2个特征
y = [[0]]
model.fit(X, y, epochs=1, batch_size=1)
```
这个例子展示了如何使用Keras创建一个简单的LSTM模型,并用一个样本进行训练。实际应用中,我们需要更多的数据和更复杂的模型来获得更好的效果。
阅读全文