我正在处理 Keras 中的不平衡数据集,并希望对我的少数类的样本给予更高的权重。该fit()
函数有一个很好的sample_weight
论点,但是由于我的数据量很大,我必须使用fit_generator()
.
fit_generator()
有一个class_weight
论点,这似乎对这个目的很有用,并且已经在另一个问题中讨论过。但是,在这种情况下,标签不是单热编码/分类的,我找不到 using 是否class_weight
也允许分类数据。
可以将class_weight
参数用于单热编码/分类标签,如果可以,如何?还是我必须求助于自定义加权损失函数?