python - 是否可以在不同的 Keras 回调中设置值？

Question

我有自定义回调，可以在某些条件下重置模型权重。但这会干扰 EarlyStopping 回调保存最佳监控值，即“val_loss”。如何访问从自定义回调到 EarlyStopping 回调的属性，以将其最佳“val_loss”设置为我的值？

Answer 1

设置你的早期停止器:

early_stopping_callback.best = new_value
early_stopping_callback.wait = 0

基于source code, currently at line 816

<小时/>

将早期塞子放入内部:

class CustomCallback(...):
    def __init__ (....., early_stopper, ...)
        self.early_stopping_callback = early_stopper

    .......

early_stopping = EarlyStopping(...)
custom = CustomCallback(...., early_stopping, ...)

您仍然需要将两个回调传递给 fit:

model.fit(..., callbacks=[custom, early_stopping])