0

需要一些帮助来为每一层的以下 LSTM 模型捕获隐藏状态和/或单元状态

我可以拼凑在一起的python代码摘录:

model = Sequential()

model.add(LSTM(units=50, return_sequences=True, return_states=True, batch_input_shape=(10,features_set.shape[1], 5),stateful=True))

model.add(Dropout(0.2))

model.add(LSTM(units=50,return_sequences=True, return_states=True,stateful=True))
model.add(Dropout(0.2))

model.add(LSTM(units=50,return_sequences=True, return_states=True,stateful=True))
model.add(Dropout(0.2))

model.add(LSTM(units=50, stateful=True))
model.add(Dropout(0.2))

model.add(Dense(units = 1))

model.compile(optimizer = 'adam', loss = 'mean_squared_error',metrics=['accuracy'])

history = model.fit(features_set, labels, epochs = 10, batch_size = 10, verbose=2)

如何保存每一层的隐藏状态和/或单元状态?到目前为止,我在网上可以找到的所有示例都只使用了一层。

请帮忙。

4

0 回答 0