阅读(7.1k) 书签 (0)

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

2019-03-25 18:19 更新

tf.keras.callbacks.CSVLogger函数

类 CSVLogger

继承自: Callback

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

将epoch(迭代次数)结果流式传输到csv文件的回调。

支持所有可以表示为字符串的值,包括一维迭代,如np.ndarray。

示例:

csv_logger = CSVLogger('training.log')
model.fit(X_train, Y_train, callbacks=[csv_logger])

参数:

  • filename:csv文件的文件名,例如:'run / log.csv'。
  • separator:用于分隔csv文件中的元素的字符串。
  • append:True:如果文件存在则追加(对继续培训很有用)。False:覆盖现有文件。

__init__

__init__(
    filename,
    separator=',',
    append=False
)

初始化自我。

方法

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)