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