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

Question

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

Answer 1

是的，它应该可以工作。班级权重是一张图表，指示班级之间的相对权重，并且在训练过程中对样本结果进行称重时会予以考虑。