keras自定义监控变量

keras自带的监控变量就是mse,mae等,有时我们想自定义监控的变量,比如学习率,或者其它一些我们自定义的参数,keras自定义监控变量和自定义损失函数很类似,都是通过这个类似内部函数的方式,例如监控学习率。

# 监控学习率
def get_lr_metric(optimizer):
    def lr(y_true,y_pred):
        return optimizer.lr
    return lr

class LRCallback(Callback):
    def on_epoch_end(self, epoch, logs=None):
        lr = self.model.optimizer.lr
        decay = self.model.optimizer.decay
        iterations = self.model.optimizer.iterations
        lr_with_decay = lr / (1. + decay * K.cast(iterations, K.dtype(decay)))
        return K.eval(lr_with_decay)

optimizer = keras.optimizers.adam(lr=1e-3,decay=0.02)
model.compile(loss = model_dice,optimizer= optimizer,metrics=['mse','mae',lr_metric])

learning_rate_callback = LRCallback()
    callbacks_list = [checkpoint,learning_rate_callback]
    history = model.fit(...,callbacks = callbacks_list)

注意通过这种嵌套函数方式,且内部函数的输入参数为y_true,y_pred,实际可能没用。

另外返回的必须是一个数字,如果想有多个返回值,我试了不可以,如果有知道的朋友请不吝赐教。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Neil_baby

你的鼓励是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值