python - 每个时期后的自定义回调以记录某些信息
问题描述
我知道如何在每个时代之后保存模型:
savemodel = ModelCheckpoint(filepath='models/model_{epoch:02d}-{loss:.2f}.h5')
model.fit(X, Y, batch_size=4, epochs=32, verbose=1, callbacks=[savemodel])
如何使用自定义回调函数来记录某些信息:
def write_metrics():
with open('log.txt', 'a') as f: # append to the log file
f.write('{epoch:02d}: loss = {loss:.1f}')
model.fit(X, Y, batch_size=4, epochs=32, verbose=1, callbacks=[savemodel, write_metrics])
?
使用此代码将无法正常工作,因为{loss}
并且{epoch}
未在f.write('{epoch:02d}: loss = {loss:.1f}')
.
解决方案
这是解决方案,通过子类化Callback
:
from keras.callbacks import Callback
class MyLogger(Callback):
def on_epoch_end(self, epoch, logs=None):
with open('log.txt', 'a+') as f:
f.write('%02d %.3f\n' % (epoch, logs['loss']))
然后
mylogger = MyLogger()
model.fit(X, Y, batch_size=32, epochs=32, verbose=1, callbacks=[mylogger])
甚至
model.fit(X, Y, batch_size=32, epochs=32, verbose=1, callbacks=[MyLogger()])
推荐阅读
- flutter - 如何在appBar中为flutter中的gridview列表实现搜索栏
- python - 在python中查找随机列表的浮点平均值
- python - 如何将for循环内函数的生成结果存储到Python中的DataFrame?
- javascript - 单击汉堡包时侧边栏而不是下拉菜单(引导程序)
- php - 为什么 if elseif 失败了?
- python - 1e-11 的 Python 浮点精度的奇怪行为
- json - 获取几个字段并遍历数组
- php - 使用 Laravel 7 上传文件时出现 FTP 连接问题
- c# - 使用 Entity Framework Core 从 Razor 视图更新列表
- eps - 如何在封装后的脚本中绘制椭圆?