Spec-Zone.ru › TensorFlow 2.9

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 если экспорт пропущен.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/estimator/BestExporter

Spec-Zone.ru

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