Keras类权重

Question

keras文档对class_weight中的model.fit()参数有点含糊。我有一个带有二进制标签的数据集，但每个类的实例不平衡。 Class 0有1000个实例，而class 1有15,000个实例。在这种情况下如何设置class_weight词典？是像class_weight = {0: 0.07, 1: 0.93}还是有其他方法来计算这些值？

Answer 1

如果将类权重设置为{0：0.07，1：0.93}，则表示模型要进行正确的类分类：1重要15倍。您可以考虑一下，因为您将只使用同一训练实例15次来训练模型。