Python tf.keras.losses.get用法及代碼示例

檢索 Keras 損失作為 function /Loss 類實例。

用法

tf.keras.losses.get(
    identifier
)

identifier 可能是損失函數或Loss 類的字符串名稱。

loss = tf.keras.losses.get("categorical_crossentropy")
type(loss)
<class 'function'>
loss = tf.keras.losses.get("CategoricalCrossentropy")
type(loss)
<class '...keras.losses.CategoricalCrossentropy'>

您還可以通過傳遞包含 class_name 和 config 作為標識符的 dict 來指定該函數的損失的 config。另請注意，class_name 必須映射到 Loss 類

identifier = {"class_name":"CategoricalCrossentropy",
              "config":{"from_logits":True} }
loss = tf.keras.losses.get(identifier)
type(loss)
<class '...keras.losses.CategoricalCrossentropy'>