tf.keras.callbacks.LambdaCallback
| Просмотреть исходный код на GitHub |
Обработчик для создания простых пользовательских обратных вызовов на лету.
Наследуется от: 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
)
Этот обратный вызов создается с анонимными функциями, которые будут вызываться в соответствующее время. Обратите внимание, что обратные вызовы ожидают позиционные аргументы, как:
-
on_epoch_beginиon_epoch_endожидают два позиционных аргумента:epoch,logs -
on_batch_beginиon_batch_endожидают два позиционных аргумента:batch,logs -
on_train_beginиon_train_endожидают один позиционный аргумент:logs
| Аргументы | |
|---|---|
on_epoch_begin | вызывается в начале каждой эпохи. |
on_epoch_end | вызывается в конце каждой эпохи. |
on_batch_begin | вызывается в начале каждой партии. |
on_batch_end | вызывается в конце каждой партии. |
on_train_begin | вызывается в начале обучения модели. |
on_train_end | вызывается в конце обучения модели. |
Пример:
# 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])
Методы
on_batch_begin
on_batch_begin(
batch, logs=None
)
Псевдоним обратной совместимости для on_train_batch_begin.
on_batch_end
on_batch_end(
batch, logs=None
)
Псевдоним обратной совместимости для on_train_batch_end.
on_epoch_begin
on_epoch_begin(
epoch, logs=None
)
Вызывается в начале эпохи.
Подклассы должны переопределять для выполнения любых действий. Эта функция должна вызываться только в режиме TRAIN.
| Аргументы | |
|---|---|
epoch | целое число, номер эпохи. |
logs | словарь. В настоящее время данные не передаются в этот аргумент для этого метода, но это может измениться в будущем. |
on_epoch_end
on_epoch_end(
epoch, logs=None
)
Вызывается в конце эпохи.
Подклассы должны переопределять для выполнения любых действий. Эта функция должна вызываться только в режиме TRAIN.
| Аргументы | |
|---|---|
epoch | целое число, номер эпохи. |
logs | словарь, результаты метрик для этой эпохи обучения и для эпохи валидации, если выполняется валидация. Ключи результатов валидации имеют префикс val_. |
on_predict_batch_begin
on_predict_batch_begin(
batch, logs=None
)
Вызывается в начале партии в методах predict.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
batch | целое число, индекс партии в текущей эпохе. |
logs | словарь. Имеет ключи batch и size, представляющие текущий номер партии и размер партии. |
on_predict_batch_end
on_predict_batch_end(
batch, logs=None
)
Вызывается в конце партии в методах predict.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
batch | целое число, индекс партии в текущей эпохе. |
logs | словарь. Результаты метрик для этой партии. |
on_predict_begin
on_predict_begin(
logs=None
)
Вызывается в начале предсказания.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
logs | словарь. В настоящее время данные не передаются в этот аргумент для этого метода, но это может измениться в будущем. |
on_predict_end
on_predict_end(
logs=None
)
Вызывается в конце предсказания.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
logs | словарь. В настоящее время данные не передаются в этот аргумент для этого метода, но это может измениться в будущем. |
on_test_batch_begin
on_test_batch_begin(
batch, logs=None
)
Вызывается в начале партии в методах evaluate.
Также вызывается в начале партии валидации в методах fit , если данные валидации предоставлены.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
batch | целое число, индекс партии в текущей эпохе. |
logs | словарь. Имеет ключи batch и size , представляющие текущий номер партии и размер партии. |
on_test_batch_end
on_test_batch_end(
batch, logs=None
)
Вызывается в конце партии в методах evaluate.
Также вызывается в конце партии валидации в методах fit , если данные валидации предоставлены.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
batch | целое число, индекс партии в текущей эпохе. |
logs | словарь. Результаты метрик для этой партии. |
on_test_begin
on_test_begin(
logs=None
)
Вызывается в начале оценки или валидации.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
logs | словарь. В настоящее время данные не передаются в этот аргумент для этого метода, но это может измениться в будущем. |
on_test_end
on_test_end(
logs=None
)
Вызывается в конце оценки или валидации.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
logs | словарь. В настоящее время данные не передаются в этот аргумент для этого метода, но это может измениться в будущем. |
on_train_batch_begin
on_train_batch_begin(
batch, logs=None
)
Вызывается в начале партии обучения в методах fit.
Подклассы должны переопределять для выполнения любых действий.
| Аргументы | |
|---|---|
batch | целое число, индекс партии в текущей эпохе. |
logs | словарь. Имеет ключи batch и size , представляющие текущий номер партии и размер партии. |
on_train_batch_end
on_train_batch_end(
batch, logs=None
)
Вызывается в конце пакетной обработки обучения в методах fit.
Подклассы должны переопределять этот метод для выполнения любых действий.
| Аргументы | |
|---|---|
batch | целое число, индекс пакета в текущей эпохе. |
logs | dict. Результаты метрики для этого пакета. |
on_train_begin
on_train_begin(
logs=None
)
Вызывается в начале обучения.
Подклассы должны переопределять этот метод для выполнения любых действий.
| Аргументы | |
|---|---|
logs | dict. В настоящее время этот аргумент не содержит данных для этого метода, но это может измениться в будущем. |
on_train_end
on_train_end(
logs=None
)
Вызывается в конце обучения.
Подклассы должны переопределять этот метод для выполнения любых действий.
| Аргументы | |
|---|---|
logs | dict. В настоящее время этот аргумент не содержит данных для этого метода, но это может измениться в будущем. |
set_model
set_model(
model
)
set_params
set_params(
params
)
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/callbacks/LambdaCallback