时间戳

首页 > 解决方案 > 创建一个每 n 个 epoch 激活的 Keras 回调

python - 创建一个每 n 个 epoch 激活的 Keras 回调

问题描述

我想在 Keras 中创建自己的回调，但并不真正了解如何去做。

我想要做的是创建一个回调，每个n（例如n=10）epochs 调用一个函数。根据 Keras 文档 ( https://keras.io/callbacks/ )，基类keras.callbacks.Callback有一个属性params。

此属性是否包括当前纪元？如果是这样，如何使用/调用它？

或者有没有其他好方法可以在每个n时期调用一个函数？

我想到了一个 Callback ，它的工作方式与keras.callbacks.ModelCheckpoint使用参数时类似period=10。

帮助将不胜感激。谢谢 ：）

编辑：经过相当多的阅读后，我想出了这个，它似乎有效（仍然需要正确测试）

class Test1(callbacks.Callback):
    def on_epoch_end(self, epochs, logs={})
        if epochs == 10:
            print('abc') #a random function

callb = Test1()    
model = networks.compute_network(layers=layers, batch_size=batch_size, epochs=epochs, call_list=[callb])
# compute_network in my case loads all the data, trains the network, and then returns it

一个网站特别帮助我更好地理解回调：https ://keunwoochoi.wordpress.com/2016/07/16/keras-callbacks/

标签： pythoncallbackkeras

解决方案

正如@HMK 所说，on_epoch_begin方法on_epoch_end提供了当前的时代。例如，要每 10 个 epoch 应用一个函数，您可以像这样编辑自定义回调：

class myCallback(keras.callbacks.Callback):
    def on_epoch_end(self, epoch, logs=None)
        if epoch % 10 == 0:
            your_func()

推荐阅读