keras中二进制分类的类不平衡

Question

我正在进行二元分类。我的y_train就像[0,1,1,0,0]。它是一维的。我在最后一层使用sigmoid作为激活，在最后一层也只使用1个输出神经元。我可以使用class_weight.compute_class_weight从y_train中查找类权重，然后将其提供给keras.fit函数中的class_weight参数。这会有用吗？

Answer 1

是的，它应该工作。类别权重是指示类别之间的相对权重的地图，并且在训练期间称量样本结果时被考虑。