Python tf.keras.callbacks.LambdaCallback用法及代碼示例

即時創建簡單的自定義回調的回調。

繼承自：Callback

用法

tf.keras.callbacks.LambdaCallback(
    on_epoch_begin=None, on_epoch_end=None, on_batch_begin=None, on_batch_end=None,
    on_train_begin=None, on_train_end=None, **kwargs
)

此回調由匿名函數構造，這些函數將在適當的時間(在 Model.{fit | evaluate | predict} 期間)調用。請注意，回調需要位置參數，如：

• on_epoch_begin 和 on_epoch_end 需要兩個位置參數：epoch , logs
• on_batch_begin 和 on_batch_end 需要兩個位置參數：batch , logs
• on_train_begin 和 on_train_end 需要一個位置參數：logs

例子：

# Print the batch number at the beginning of every batch.
batch_print_callback = LambdaCallback(
    on_batch_begin=lambda batch,logs:print(batch))

# Stream the epoch loss to a file in JSON format. The file content
# is not well-formed JSON but rather has a JSON object per line.
import json
json_log = open('loss_log.json', mode='wt', buffering=1)
json_logging_callback = LambdaCallback(
    on_epoch_end=lambda epoch, logs:json_log.write(
        json.dumps({'epoch':epoch, 'loss':logs['loss']}) + '\n'),
    on_train_end=lambda logs:json_log.close()
)

# Terminate some processes after having finished model training.
processes = ...
cleanup_callback = LambdaCallback(
    on_train_end=lambda logs:[
        p.terminate() for p in processes if p.is_alive()])

model.fit(...,
          callbacks=[batch_print_callback,
                     json_logging_callback,
                     cleanup_callback])