tf.keras.callbacks.EarlyStopping
| Просмотреть исходный код на GitHub |
Прекратить обучение, когда отслеживаемая величина перестанет улучшаться.
Наследуется от: Callback
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