lstm神经网络python代码
时间: 2023-09-15 11:23:12 浏览: 98
LSTM Python代码
5星 · 资源好评率100%
以下是一个简单的LSTM神经网络的Python代码示例:
```python
from keras.models import Sequential
from keras.layers import LSTM, Dense
import numpy as np
# 创建模型
model = Sequential()
# 添加一个LSTM层
model.add(LSTM(units=64, activation='tanh', input_shape=(100, 1)))
# 添加一个全连接层
model.add(Dense(units=1))
# 编译模型
model.compile(loss='mean_squared_error', optimizer='adam')
# 生成模拟数据
data = np.random.rand(1000, 100, 1)
labels = np.random.rand(1000, 1)
# 训练模型
model.fit(data, labels, epochs=10, batch_size=32)
```
这个代码创建了一个具有一个LSTM层和一个全连接层的模型。我们使用随机生成的数据和标签进行训练。在实际应用中,您需要准备自己的数据集并进行训练。
阅读全文