首页 > 解决方案 > LSTM中记忆单元与时间步长的关系

问题描述

我正在研究 LSTM 模型。

LSTM中隐藏层的一个记忆单元是否对应一个时间步长?

示例代码) model.add(LSTM(128, input_shape = (4, 1)))

在 Keras 中实现 LSTM 时,可以设置存储单元的数量,如示例代码中所示,无论时间步长如何。在示例中为 128。

但是,一个典型的 LSTM 图像显示为与时间步数和记忆单元数 1:1 对应。正确答案是什么?

在此处输入图像描述

标签: keraslstmtimestep

解决方案


在 LSTM 中,我们以下列方式提供输入 [samples,timesteps,features] samples 用于您想要在某个时间提供的训练示例的数量 timesteps 是您想要使用的值多少假设您提到 timesteps=3 所以值在 t ,t-1 和 t-2 用于预测 t+1 处的数据特征是您希望一次提供多少维 LSTM 有记忆单元但我正在解释代码部分以免混淆你我希望这个帮助


推荐阅读