Keras 验证期间的类加权

数据挖掘 神经网络 分类 喀拉斯 阶级失衡
2021-10-10 16:17:03

我想知道类权重是否也用于评估训练期间验证数据的损失。如果没有,有没有办法调整 fit() 函数,以便在验证损失计算期间考虑类权重。

1个回答

Keras.models.py[1] 中所述,validation_data可以是 tuple (inputs, targets, sample_weights)因此,您可以在验证损失计算期间使用此参数进行类加权。


[1] https://github.com/fchollet/keras/blob/master/keras/models.py