首页 > 解决方案 > Keras:是否有具有历史记录+进度的 train_on_batch 示例代码?

问题描述

在 Python 中,我正在从 KerasModel.fit转移到Model.train_on_batch循环以获得更好的控制。但是返回的进度条和 History 对象fit很有用。在浪费时间从头开始实现它们之前,我想知道是否有人找到train_on_batch了使用重现进度条和历史记录的示例代码?

(注意。我查看了 的源代码fit但是有足够多的间接层,很难准确地挖掘出它在做什么。还发现了这个,这很有帮助,但没有相关的功能。)

标签: pythontensorflowkeras

解决方案


因此,在查看了 keras 的源代码后,我发现tf.keras.callbacks.ProgbarLoggertf.keras.callbacks.History就是你想要的

源代码

keras/callbacks.py#L259

keras/callbacks.py#L263


推荐阅读