tf.keras.utils.custom_object_scope

Provides a scope that changes to _GLOBAL_CUSTOM_OBJECTS cannot escape.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.utils.custom_object_scope, `tf.compat.v2.keras.utils.custom_object_scope`

tf.keras.utils.custom_object_scope(
    *args
)

Convenience wrapper for CustomObjectScope. Code within a with statement will be able to access custom objects by name. Changes to global custom objects persist within the enclosing with statement. At end of the with statement, global custom objects are reverted to state at beginning of the with statement.

Example:

Consider a custom object MyObject

with custom_object_scope({'MyObject':MyObject}):
    layer = Dense(..., kernel_regularizer='MyObject')
    # save, load, etc. will recognize custom object by name

Arguments
*args	Variable length list of dictionaries of name, class pairs to add to custom objects.

Returns
Object of type CustomObjectScope.