python - 使用 Keras 深度学习的不平衡数据集

Question

我有一个如下所示的数据集：Training (Class 0: 471, Class 1: 986) Testing (Class 0: 177, Class 1: 246。我将数据分成 80% 用于训练，20% 用于验证。我知道那是一个不平衡的数据集，我已经尝试过 Class_weight 但问题仍然存在。我重新训练了我的基线 CNN，我总是得到如图所示的结果。有人可以帮我吗？

Answer 1

在将事件分类为 5 个不平衡的类别时，我遇到了类似的问题。我发现这个实现加权分类交叉熵的损失函数：https ://gist.github.com/noparade/aaa8584e6e90ad64936e333e4e08ca5f 结合 Nadam 优化器，它让我对所有类别都获得了超过 95% 的真阳性。