Spec-Zone.ru › TensorFlow 1.15

tf.keras.callbacks.EarlyStopping

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

Прекратить обучение, когда отслеживаемая величина перестанет улучшаться.

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

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

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

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

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

tf.keras.callbacks.EarlyStopping(
    monitor='val_loss', min_delta=0, patience=0, verbose=0, mode='auto',
    baseline=None, restore_best_weights=False
)
Аргументы
monitor Величина, которая должна отслеживаться.
min_delta Минимальное изменение в отслеживаемой величине, которое квалифицируется как улучшение, т. е. абсолютное изменение меньше min_delta будет считаться отсутствием улучшения.
patience Количество эпох без улучшения, после которого обучение будет остановлено.
verbose Режим отображения.
mode Один из {"auto", "min", "max"}. В режиме min обучение остановится, когда отслеживаемая величина перестанет убывать; в режиме max обучение остановится, когда отслеживаемая величина перестанет возрастать; в режиме auto направление автоматически определяется по имени отслеживаемой величины.
baseline Базовое значение для отслеживаемой величины. Обучение остановится, если модель не показывает улучшения по сравнению с базовым значением.
restore_best_weights Восстановить ли веса модели из эпохи с лучшим значением отслеживаемой величины. Если False, используются веса модели, полученные на последнем шаге обучения.

Пример:

callback = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=3)
# This callback will stop the training when there is no improvement in
# the validation loss for three consecutive epochs.
model.fit(data, labels, epochs=100, callbacks=[callback],
    validation_data=(val_data, val_labels))

Методы

get_monitor_value

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

get_monitor_value(
    logs
)

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

on_train_begin

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

on_train_begin(
    logs=None
)

Вызывается в начале обучения.

Подклассы должны переопределять этот метод для выполнения любых действий.

Аргументы
logs словарь. В данный момент для этого метода данные не передаются в этот аргумент, но это может измениться в будущем.

on_train_end

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

on_train_end(
    logs=None
)

Вызывается в конце обучения.

Подклассы должны переопределять этот метод для выполнения любых действий.

Аргументы
logs словарь. В данный момент для этого метода данные не передаются в этот аргумент, но это может измениться в будущем.

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/EarlyStopping

Spec-Zone.ru

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