tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater

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

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

Класс MethodNameUpdater предоставляет функциональность для обновления поля имени метода в signature_defs заданной сохранённой модели. Например, его можно использовать для замены 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.

Методы

replace_method_name

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

replace_method_name(
    signature_key, method_name, tags=None
)

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

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

save

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

save(
    new_export_dir=None
)

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