阅读(7.3k) 书签 (0)

TensorFlow回调函数:tf.keras.callbacks.Callback

2019-03-25 18:18 更新

tf.keras.callbacks.Callback函数

类 Callback

定义在:tensorflow/python/keras/callbacks.py。

用于构建新回调的抽象基类。

属性:

  • params:字典。训练参数(例如,详细程度,批量大小,epoch数......)。
  • model:keras.models.Model的实例。正在训练的模型的参考。

被回调方法作为参数的logs字典将包含与当前批次或epoch相关的数量的键。

目前,Sequential模型类的.fit()方法会在传入到回调函数的logs里面包含以下的数据:

  • on_epoch_end:包括acc和loss的日志,并且可选地包括val_loss (如果在fit中启用了验证),以及val_acc(如果启用了验证和准确性监视)。
  • on_batch_begin:logs包括size,当前批次中的样本数。
  • on_batch_end:日志包括loss,也可以选择acc (如果启用了准确性监控)。

__init__

__init__()

初始化自我。

方法

on_batch_begin

on_batch_begin(
    batch,
    logs=None
)

on_batch_end

on_batch_end(
    batch,
    logs=None
)

on_epoch_begin

on_epoch_begin(
    epoch,
    logs=None
)

on_epoch_end

on_epoch_end(
    epoch,
    logs=None
)

on_train_batch_begin

on_train_batch_begin(
    batch,
    logs=None
)

on_train_batch_end

on_train_batch_end(
    batch,
    logs=None
)

on_train_begin

on_train_begin(logs=None)

on_train_end

on_train_end(logs=None)

set_model

set_model(model)

set_params

set_params(params)