Python tf.keras.utils.get_registered_object用法及代碼示例

如果在 Keras 中注冊，則返回與 name 關聯的類。

用法

tf.keras.utils.get_registered_object(
    name, custom_objects=None, module_objects=None
)

參數

name 要查找的名稱。
custom_objects 用於查找名稱的自定義對象字典。通常，custom_objects 由用戶提供。
module_objects 用於查找名稱的自定義對象字典。通常，module_objects 由中級庫實現者提供。

與'name' 關聯的可實例化類，如果不存在此類，則為 None。

該函數是 Keras 序列化和反序列化框架的一部分。它將字符串映射到與其關聯的對象以進行序列化/反序列化。

例子：

def from_config(cls, config, custom_objects=None):
  if 'my_custom_object_name' in config:
    config['hidden_cls'] = tf.keras.utils.get_registered_object(
        config['my_custom_object_name'], custom_objects=custom_objects)

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.utils.get_registered_object。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。