【发布时间】:2021-08-29 05:02:00
【问题描述】:
我正在尝试编写一个自定义回调,该回调将在 epoch 结束时更新样本权重。我用原始权重初始化自定义回调,但我不确定如何确保 keras 使用回调中定义的新样本权重来拟合模型。这是我的代码的一个简单示例(* 2 是一个示例,实际上不应该做任何事情)。
class update_weights(tf.keras.callbacks.Callback):
def __init__(self, sample_weight):
self.sample_weight = sample_weight
def on_epoch_end(self, epoch, logs={}):
self.sample_weight = self.sample_weight * 2
似乎可以从简单的 self.model 访问模型参数。但是我很难访问 fit 函数的参数。甚至可能吗?我可以在校准模型时修改拟合函数的参数吗?
【问题讨论】:
标签: python tensorflow keras weighted