我添加了以下代码来添加分类器,结果获取了一个低置信度分数。
我使用以下代码训练了模型,使用2048张图像,分类为猫和狗两类
history = model.fit(X_train, Y_train, epochs=5, batch_size=32, validation_data=(X_test, Y_test))
回答:
在解决二分类问题时,如果你使用sigmoid作为激活函数,你的密集层应该有一个神经元而不是两个。如果你使用softmax,那么你的密集层应该有两个神经元。你应该在这两者之间选择,并相应地更改输出神经元的数量。你可以在这里了解更多这里。