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

檢索 Keras 指標作為 function /Metric 類實例。

用法

tf.keras.metrics.get(
    identifier
)

identifier 可能是度量函數或類的字符串名稱。

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

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

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