tf.keras.callbacks.LearningRateScheduler

Планировщик скорости обучения.

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

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

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

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

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

tf.keras.callbacks.LearningRateScheduler(
    schedule, verbose=0
)

Аргументы
`schedule`	функция, принимающая индекс эпохи в качестве входных данных (целое число, индексируется с 0) и возвращающая новую скорость обучения в качестве выходных данных (вещественное число).
`verbose`	целое число. 0: без вывода, 1: выводить сообщения об обновлениях.

# This function keeps the learning rate at 0.001 for the first ten epochs
# and decreases it exponentially after that.
def scheduler(epoch):
  if epoch < 10:
    return 0.001
  else:
    return 0.001 * tf.math.exp(0.1 * (10 - epoch))

callback = tf.keras.callbacks.LearningRateScheduler(scheduler)
model.fit(data, labels, epochs=100, callbacks=[callback],
          validation_data=(val_data, val_labels))

Методы

`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
)

Вызывается в конце эпохи.

Аргументы
`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`	словарь. Результаты метрики для этого пакета.

`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/LearningRateScheduler