Tensorflow始终预测为0

Question

我试图训练NN进行二进制分类但使用多类方法，因此Class（C）= 2。列车和测试数据集的准确度约为96％。但是，手动检查显示Tensorflow始终为每个示例选择0。

我有兴趣了解如何调试此问题以及我做错了什么。非常感谢您的帮助。感谢。

（如果我过于模糊并且没有提供足够的信息以便您找到一个好的解决方案，请告诉我。谢谢。）

Answer 1

如果您的准确率为96％且所有预测均为0，则您的数据集可能不平衡。您应该平衡它并使正负类中的样本数相同。您还可以为该类添加权重，使用较少的样本惩罚更多的类。