【发布时间】:2021-02-10 20:25:58
【问题描述】:
我如何用 Keras 绘制一个用 python 制作的 LSTM 网络。网络接收 6 个不同的参数作为输入,并返回一个预测值。神经网络设置如下:
model = Sequential()
model.add(LSTM(512,input_shape=(train_X.shape[1], train_X.shape[2])))
model.add(Dense(1))
model.compile(loss='mae', optimizer='adam')
model.summary()
Model: "sequential_3"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
lstm_3 (LSTM) (None, 512) 1062912
_________________________________________________________________
dense_3 (Dense) (None, 1) 513
=================================================================
Total params: 1,063,425
Trainable params: 1,063,425
Non-trainable params: 0
下面的图片会是一个很好的表示吗?
如果图像不正确,我可以使用正常神经元的表示来代替 LSTM 细胞吗?
【问题讨论】:
标签: python machine-learning keras