Skip to main content
 首页 » 编程设计

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

2025年05月04日15leader

我有一个具有相同数量标签的文本数据集 - 0,1,2,3,4。我在他们的网站上用我的数据集运行了 Keras 二进制分类 示例 LSTM(imdb 示例),编译行更改为

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

但该模型仅预测 1 个类别,即 20% 的准确率始终如一。

你能帮我修复它/根据需要更改设置吗?

请您参考如下方法:

你需要修改 with

model.add(Dense(nb_classes)) 

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