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

Question

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

Answer 1

它不是在Keras的文档中，而是经常使用的层具有.reset_state（）函数。你可以在github上找到它，你可以重置一层或整个模型的状态。

Answer 2

在Keras文档，Chapter on Recurrent Layers：

要重置模型的状态，呼叫.reset_states（）是一个特定层上，或者在整个模型。

例如为：

model.predict() 

model.reset_states()