tf.keras.utils.register_keras_serializable

Регистрирует объект в системе сериализации Keras.

tf.keras.utils.register_keras_serializable(
    package='Custom', name=None
)

Этот декоратор добавляет декорированный класс или функцию в словарь пользовательских объектов Keras, чтобы их можно было сериализовать и десериализовать без добавления записи в предоставленный пользователем словарь пользовательских объектов. Он также добавляет функцию, которую Keras будет вызывать для получения сериализуемого строкового ключа объекта.

Обратите внимание: чтобы классы можно было сериализовать и десериализовать, они должны реализовать метод get_config(). Для функций такого требования нет.

Объект будет зарегистрирован под ключом 'package>name', где name. Если имя не указано, по умолчанию используется имя объекта.

Пример:

# Note that `'my_package'` is used as the `package` argument here, and since
# the `name` argument is not provided, `'MyDense'` is used as the `name`.
@register_keras_serializable('my_package')
class MyDense(keras.layers.Dense):
    pass

assert get_registered_object('my_package>MyDense') == MyDense
assert get_registered_name(MyDense) == 'my_package>MyDense'