tf.estimator.FinalExporter
| Просмотреть исходный код на GitHub |
Этот класс экспортирует служебную граф и контрольные точки в конце.
Наследуется от: Exporter
tf.estimator.FinalExporter(
name, serving_input_receiver_fn, assets_extra=None, as_text=False
)
Этот класс выполняет одно экспортное действие в конце обучения.
| Аргументы | |
|---|---|
name | уникальное имя этого Exporter, которое будет использоваться в пути экспорта. |
serving_input_receiver_fn | функция, которая не принимает аргументов и возвращает ServingInputReceiver. |
assets_extra | Необязательный словарь, определяющий способ заполнения каталога assets.extra в экспортированной SavedModel. Каждый ключ должен указывать путь назначения (включая имя файла) относительно каталога assets.extra. Соответствующее значение указывает полный путь исходного файла, который будет скопирован. Например, простой случай копирования одного файла без переименования задаётся как {'my_asset_file.txt': '/path/to/my_asset_file.txt'}. |
as_text | следует ли записывать протокол SavedModel в текстовом формате. По умолчанию False. |
| Исключения | |
|---|---|
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/FinalExporter