具有极不平衡数据集的Tensorflow分类

Question

我使用TensorFlow LinearClassifier和DNN来分类两类数据集.

但是,问题是数据集包含96%的正输出和4%的负输出,我的程序总是将预测返回为正数.当然,在这种情况下,我将达到96%的准确率,但它根本没有意义.

处理这种情况的好方法是什么？

Answer 1

您可以尝试更改成本函数，以便误报输出比误报输出受到更严重的惩罚。