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'