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)