machine-learning - 你能让 LSTM 手动忘记上下文吗？

Question

我对机器学习非常陌生，并且想知道是否可以手动清空 LSTM 的短期内存。举例来说，我想在句子上训练 LSTM“ jack 和吉尔上了，”但决定不结束这句话。如果我想用“Humpty dumpty sat on a wall”来训练它，我该如何防止它立即预测“hill”这个词？我正在使用 Keras。

Answer 1

相关:When does keras reset an LSTM state?

如果您的 LSTM 是无状态的(stateful = False，我认为这对您来说更有意义)，或者如果您通过将两者都放在这两个句子上来训练 LSTM，则不需要重置上下文同一批。

如果您确实这样做，请使用model.reset_state()。