我有一个具有相同数量标签的文本数据集 - 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 对应于分类类的数量。
