Keras 3 API 文件 / 工具程式 / 實驗管理工具程式

實驗管理工具程式

[原始碼]

Config 類別

keras.utils.Config(**kwargs)

Config 是一個類似字典的容器,用於存放具名值。

相較於普通的字典,它提供了一些優勢

  • 透過屬性設定/獲取來設定和檢索值。
  • 能夠凍結配置,以確保在程式碼的特定點之後不會發生意外的配置修改。
  • 輕鬆將整個配置序列化為 JSON。

範例

# Create a config via constructor arguments
config = Config("learning_rate"=0.1, "momentum"=0.9)

# Then keep adding to it via attribute-style setting
config.use_ema = True
config.ema_overwrite_frequency = 100

# You can also add attributes via dict-like access
config["seed"] = 123

# You can retrieve entries both via attribute-style
# access and dict-style access
assert config.seed == 100
assert config["learning_rate"] == 0.1

Config 的行為類似字典

config = Config("learning_rate"=0.1, "momentum"=0.9)
for k, v in config.items():
    print(f"{k}={v}")

print(f"keys: {list(config.keys())}")
print(f"values: {list(config.values())}")

事實上,它可以轉換成字典

config = Config("learning_rate"=0.1, "momentum"=0.9)
dict_config = config.as_dict()

您可以輕鬆地將配置序列化為 JSON

config = Config("learning_rate"=0.1, "momentum"=0.9)

json_str = config.to_json()

您也可以凍結配置以防止進一步的更改

config = Config()
config.optimizer = "adam"
config.seed = 123

# Freeze the config to prevent changes.
config.freeze()
assert config.frozen

config.foo = "bar"  # This will raise an error.