在TensorFlow Keras API中,如何向CSVLogger回调创建的csv文件中添加参数?

问题描述

我如何创建一个TensorFlow Keras API回调,以便在每个时期将学习率值添加到tf.keras.callbacks.CSVLogger创建的csv文件中?

通过下面的回调,我可以在每个时期之后打印出我的学习率,并将其添加到历史记录中。但是我不知道如何将其添加到CSVLogger csv文件中。似乎有一个存储在CSV记录器中的值存储在地方的日志字典,但我不知道它在哪里或如何添加

class Print_lr(tf.keras.callbacks.Callback):
    def on_epoch_begin(self,epoch,logs=None):
        print('lr = %f' % self.model.optimizer.lr)
        if 'lr' not in self.model.history.history.keys():
            self.model.history.history['lr'] = []
        self.model.history.history['lr'].append(self.model.optimizer.lr.numpy())

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)