在训练期间更改keras class_weight

Question

我想利用Keras中fit函数的 class_weight 参数来处理不平衡的数据集。

关键在于我想通过基于正在运行的纪元的函数指定 class_weights 。

我虽然使用回调函数，但我无法从那里更改 class_weight 参数。有什么帮助吗？

Answer 1

此时很遗憾无法更改回调函数中的class_weight参数。该参数未在模型中保存为实例变量，因此您无法访问它。

但是，如果使用model.train_on_batch，则可以调整每个批次的类权重参数。您将能够从批号中计算时期编号。但这仅适用于这种特殊的拟合方法。因此，在将批量数据传递给fit函数之前，您需要创建一个批处理生成器并修改类权重参数。