如何在Keras中重置状态变量

Question

我正在构建一个RNN (LSTM/GRU)，我有可变长度的时间序列。如何在每个时间序列完成后重置状态变量?我想在Keras中做到这一点，但如果我必须在Theano(我的后端)中构建RNN，那也很好。

Answer 1

没有在Keras的文档中，但是循环层有一个.reset_state()函数。你可以在github上找到它，你可以重置一个层或整个模型的状态。

Answer 2

在Keras文档中，循环层章节:

要重置模型的状态，在特定层或整个模型上调用.reset_states()。

例如:

model.predict()
model.reset_states()