Keras LSTM在多类别分类中仅预测1个类别 - 如何修复？

Question

我有一个文本数据集，其标签数量相同 - 0,1,2,3,4。我使用我的数据集在他们的网站上运行了Keras binary classification示例LSTM（imdb示例），编译行更改为

"model.compile(loss='categorical_crossentropy', optimizer='adam', class_mode="categorical")"

但该模型仅预测1个类别，即20%一致的准确度。

您可以根据需要帮我修理/更改设置吗？

Answer 1

您需要修改with

model.add(Dense(nb_classes))

其中nb_classes对应于分类类的数量。