如何在Keras中重置状态变量？

Question

我正在构建一个RNN（LSTM / GRU），我有可变长度的时间序列。每个时间序列完成后如何重置状态变量？我想在Keras这样做，但如果我必须在Theano（我的后端）建立一个RNN，那也没关系。

Answer 1

这不是在Keras＆＃39;文档但循环图层有一个.reset_state（）函数。您可以在github上找到它，您可以重置一个层或整个模型的状态。

Answer 2

在Keras文档中，Chapter on Recurrent Layers：

  

要重置模型的状态，请在特定图层或整个模型上调用.reset_states（）。

E.g：

model.predict()

model.reset_states()