Spec-Zone.ru › TensorFlow 1.15

tf.keras.callbacks.ModelCheckpoint

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

Сохраняет модель после каждой эпохи.

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

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

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

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

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

tf.keras.callbacks.ModelCheckpoint(
    filepath, monitor='val_loss', verbose=0, save_best_only=False,
    save_weights_only=False, mode='auto', save_freq='epoch', **kwargs
)

filepath может содержать именованные параметры форматирования, которые будут заполнены значением epoch и ключами в logs (переданными в on_epoch_end).

Например: если filepath равно weights.{epoch:02d}-{val_loss:.2f}.hdf5, то контрольные точки модели будут сохраняться с номером эпохи и значением валидационной ошибки в имени файла.

Аргументы
filepath строка, путь для сохранения файла модели.
monitor величина для отслеживания.
verbose режим отображения, 0 или 1.
save_best_only если save_best_only=True, самая последняя лучшая модель в соответствии с отслеживаемой величиной не будет перезаписана.
mode один из {auto, min, max}. Если save_best_only=True, решение о перезаписи текущего файла сохранения принимается на основе максимизации или минимизации отслеживаемой величины. Для val_acc, это должно быть max, для val_loss это должно быть min, и т.д. В режиме auto, направление автоматически определяется по имени отслеживаемой величины.
save_weights_only если True, то будут сохранены только веса модели (model.save_weights(filepath)), иначе сохраняется вся модель (model.save(filepath)).
save_freq 'epoch' или целое число. При использовании 'epoch', обратный вызов сохраняет модель после каждой эпохи. При использовании целого числа обратный вызов сохраняет модель в конце пакета, после обработки такого-то количества образцов с момента последнего сохранения. Обратите внимание, что если сохранение не выровнено с эпохами, отслеживаемая метрика может быть менее надежной (она может отражать всего 1 пакет, поскольку метрики сбрасываются в каждой эпохе). По умолчанию 'epoch'
**kwargs Дополнительные аргументы для обратной совместимости. Возможный ключ period.

Методы

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
)

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

Подклассы должны переопределить для выполнения любых действий. Эта функция должна вызываться только в режиме обучения.

Аргументы
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 словарь. В настоящее время данные не передаются в этот аргумент для данного метода, но это может измениться в будущем.

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

Spec-Zone.ru

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