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

Posted 2023-02-16

技术标签:

【中文标题】如何在 Keras 中重置状态变量？【英文标题】：How to reset state variables in Keras? 【发布时间】：2016-12-21 20:21:17 【问题描述】：

我正在构建一个 RNN (LSTM/GRU)，并且我有可变长度的时间序列。每次时间序列完成后如何重置状态变量？我想在 Keras 中执行此操作，但如果我必须在 Theano（我的后端）中构建一个 RNN，那也没关系。

【问题讨论】：

【参考方案1】：

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

【讨论】：

【参考方案2】：

在 Keras 文档中，Chapter on Recurrent Layers：

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

例如：

model.predict()

model.reset_states()

【讨论】：

以上是关于如何在 Keras 中重置状态变量？的主要内容，如果未能解决你的问题，请参考以下文章