如何重置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()