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