Spec-Zone.ru › TensorFlow 2.4

tf.estimator.BestExporter

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

Этот класс экспортирует служебную графу и контрольные точки лучших моделей.

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

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

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

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

tf.compat.v1.estimator.BestExporter

tf.estimator.BestExporter(
    name='best_exporter', serving_input_receiver_fn=None,
    event_file_pattern='eval/*.tfevents.*', compare_fn=_loss_smaller,
    assets_extra=None, as_text=False, exports_to_keep=5
)

Этот класс выполняет экспорт модели каждый раз, когда новая модель лучше любой существующей.

Аргументы
name уникальное имя данного Exporter, которое будет использоваться в пути экспорта.
serving_input_receiver_fn функция, которая не принимает аргументы и возвращает ServingInputReceiver.
event_file_pattern шаблон имени файла событий, относительный к каталогу model_dir. Если None, экспортер не будет безопасным от прерываний. Для обеспечения безопасности от прерываний необходимо указать event_file_pattern.
compare_fn функция, которая сравнивает два результата оценки и возвращает true, если текущий результат оценки лучше. Следует следующему соглашению:
  • Аргументы:
  • best_eval_result: Это результат оценки лучшей модели.
  • current_eval_result: Это результат оценки текущей модели-кандидата.
  • Возвращает: True, если текущий результат оценки лучше; в противном случае False.
assets_extra Необязательный словарь, определяющий способ заполнения каталога assets.extra в экспортированной SavedModel. Каждый ключ должен указывать путь назначения (включая имя файла) относительно каталога assets.extra. Соответствующее значение указывает полный путь к исходному файлу, который необходимо скопировать. Например, простой случай копирования одного файла без переименования задается как {'my_asset_file.txt': '/path/to/my_asset_file.txt'}.
as_text флаг записи протокола SavedModel в текстовом формате. По умолчанию False.
exports_to_keep Количество экспортов для сохранения. Более старые экспорты будут удалены. По умолчанию 5. Установите в None для отключения удаления.
Возбуждает
ValueError если какой-либо аргумент некорректен.
Атрибуты
name Имя каталога.

Имя каталога в каталоге экспорта, в котором записываются экспорты данного типа. Не должно быть None и не должно быть пустым.

Методы

export

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

export(
    estimator, export_path, checkpoint_path, eval_result, is_the_final_export
)

Экспортирует заданную Estimator в определённый формат.

Аргументы
estimator экспортируемая Estimator
export_path строка, содержащая каталог для записи экспорта.
checkpoint_path путь к контрольной точке для экспорта.
eval_result результат Estimator.evaluate для данной контрольной точки.
is_the_final_export Этот булевый флаг равен True, когда это экспорт в конце обучения. Он равен False для промежуточных экспортов во время обучения. При передаче Exporter в tf.estimator.train_and_evaluate is_the_final_export всегда равен False, если TrainSpec.max_steps равен None.
Возвращает
Путь к экспортированному каталогу или None если экспорт пропущен.

© 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/r2.4/api_docs/python/tf/estimator/BestExporter

Spec-Zone.ru

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