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)