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'>