python - Keras LSTM 在每个单元上使用 softmax

Question

我正在创建一个类似于下面提到的模型： 模型

我正在使用 Keras 创建这样的模型，但由于我无法找到将 SoftMax 添加到 LSTM 单元的输出的方法而陷入了死胡同。到目前为止，所有教程和帮助材料都提供了有关输出单个类的信息，即使在此链接中提供的图像字幕的情况下也是如此。

那么是否可以将 SoftMax 应用于 LSTM 的每个单元（返回序列为真），或者我是否必须转移到 pytorch。

标签： pythonkeraslstmrecurrent-neural-network

Answer 1

答案是：是的，可以应用于 LSTM 的每个单元，不，您不必迁移到 PyTorch。

在 Keras 1.X 中，您需要明确声明您添加了一个 TimeDistributed 层，而在 Keras 2.X 中，您只需编写：

model.add(LSTM(50,activation='relu',return_sequences=False))
model.add(Dense(number_of_classes,activation='softmax'))