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

您需要与

model.add(Dense(nb_classes)) 

修改的地方nb_classes对应分类类别的数量。