LSTM 的输出在哪里？

数据挖掘 神经网络 lstm rnn
2022-03-12 03:54:26

我试图了解 LSTM 的输出在哪里。请参考下图：

http://colah.github.io/posts/2015-08-Understanding-LSTMs/

似乎在每个时间步，我们输出对应于隐藏状态和记忆单元的 h_t 和 C_t。

现在假设我正在尝试对二进制 [0,1] 的股票价格变动进行建模，0 代表下跌，1 代表上涨，这是我的 y_i。

我输入 x_t，它是每个 tilmestep 的特征向量，我希望在最后一个 tilmestep 之后获得一维输出 y_t。

h_t 是我要找的吗？这意味着 h_t 与输出维度匹配，但出于某种原因，我认为它与输出维度无关。

1个回答

通常你必须Dense在单元之后添加一个层LSTM。这将尝试了解如何使用 LSTM 的输出。

例如在 Keras 中：

model = Sequential()
model.add(LSTM(4, input_shape=(1, 5)))
model.add(Dense(1, activation='sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='adam')
其它你可能感兴趣的问题
上一篇如何生成这样的输出？ 下一篇DQN 中训练和测试的最后几集有什么区别？