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