我初始化LSTM层状态,但没有效果。那么如何初始化RNN层的状态

时间:2019-01-09 02:29:43

标签: keras state lstm

我想初始化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)

但是我仍然不知道第一个代码的错误。

0 个答案:

没有答案