tf.estimator.BestExporter
| Просмотреть исходный код на GitHub |
Этот класс экспортирует служебную граф и контрольные точки лучших моделей.
Наследуется от: Exporter
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, если текущий результат оценки лучше. Следует сигнатуре:
|
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 | Имя каталога. Имя каталога в каталоге экспорта, где пишутся экспорты этого типа. Не должно быть |
Методы
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