Python tf.keras.utils.custom_object_scope用法及代码示例

向 Keras 反序列化内部公开自定义类/函数。

用法

tf.keras.utils.custom_object_scope(
    *args
)

在 with custom_object_scope(objects_dict) 范围内，诸如 tf.keras.models.load_model 或 tf.keras.models.model_from_config 之类的 Keras 方法将能够反序列化保存的配置(例如自定义层或指标)引用的任何自定义对象。

例子：

考虑一个自定义正则化器 my_regularizer ：

layer = Dense(3, kernel_regularizer=my_regularizer)
config = layer.get_config()  # Config contains a reference to `my_regularizer`
...
# Later:
with custom_object_scope({'my_regularizer':my_regularizer}):
  layer = Dense.from_config(config)