Spec-Zone.ru › TensorFlow 1.15

tf.keras.callbacks.LambdaCallback

Просмотреть исходный код на GitHub

Обработчик для создания простых пользовательских обратных вызовов на лету.

Наследуется от: Callback

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.keras.callbacks.LambdaCallback, `tf.compat.v2.keras.callbacks.LambdaCallback`

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API