我想初始化LSTM层的状态,我找到了一些方法,但是我不知道为什么我的预测结果没有变化。
这是我的代码:
qcom-based
结果与未初始化的结果相同。
#好的,我更改了代码,它可以按预期工作。这是我的代码
new_input = Input(shape=(1, 1), batch_shape=(batch_size, 1, 1))
all_states, last_states, cell_states = LSTM(20, input_shape=(1, 1), batch_size=batch_size, return_sequences=True, return_state=True, stateful=True)(new_input)
new_model2 = Model(inputs=new_input, outputs=[all_states, last_states, cell_states])
new_model2.set_weights(lstm_weights)
new_model2.layers[1].states[0] = hidden_state
new_model2.layers[1].states[1] = cell_state
results = new_model2.predict(x_test[:3], batch_size=batch_size)
但是我仍然不知道第一个代码的错误。