tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater

Обновляет имя метода (имена методов) SavedModel, сохранённой по указанному пути.

tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater(
    export_dir
)

Класс MethodNameUpdater предоставляет возможность обновлять поле имени метода в signature_defs указанной SavedModel. Например, его можно использовать для замены predict method_name на regress.

Типичное использование MethodNameUpdater

...
updater = tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater(
    export_dir)
# Update all signature_defs with key "foo" in all meta graph defs.
updater.replace_method_name(signature_key="foo", method_name="regress")
# Update a single signature_def with key "bar" in the meta graph def with
# tags ["serve"]
updater.replace_method_name(signature_key="bar", method_name="classify",
                            tags="serve")
updater.save(new_export_dir)

Примечание: Эта функция будет доступна только через библиотеку совместимости v1 как tf.compat.v1.saved_model.builder.MethodNameUpdater.

Аргументы
`export_dir`	Каталог, содержащий файлы SavedModel.

Исключения
`IOError`	Если файл сохранённой модели не существует или его не удалось успешно разобрать.

Методы

`replace_method_name`

Посмотреть исходный код

replace_method_name(
    signature_key, method_name, tags=None
)

Заменяет method_name в указанном signature_def.

Если tags имеет значение None (то есть у нескольких MetaGraph есть signature_def с одинаковым ключом), будут найдены и заменены несколько sig def. Если tags не равно None, будет заменён только один signature_def в MetaGraph с соответствующими тегами.

Аргументы
`signature_key`	Ключ signature_def, который нужно обновить.
`method_name`	Новое имя метода, заменяющее существующее.
`tags`	Тег или последовательность тегов, идентифицирующих `MetaGraph` для обновления. Если значение равно None, будут обновлены все метаграфы.

Исключения
`ValueError`	Если signature_key или method_name не заданы, если не найдено ни одного метаграфа с соответствующими тегами или если ни в одном метаграфе нет signature_def с ключом signature_key.

`save`

Посмотреть исходный код

save(
    new_export_dir=None
)

Сохраняет обновлённую SavedModel.

Аргументы
`new_export_dir`	Путь, по которому будет сохранена обновлённая `SavedModel`. Если значение равно None, входная `SavedModel` будет перезаписана с учётом обновлений.

Исключения
`errors.OpError`	Если при сохранении файла возникли ошибки.

© 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/api_docs/python/tf/compat/v1/saved_model/signature_def_utils/MethodNameUpdater