register_keras_serializable function

Registers a custom object with the Keras serialization framework.