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

用 Mean 度量包裝一個無狀態度量函數。

繼承自：Mean、Metric、Layer、Module

用法

tf.keras.metrics.MeanMetricWrapper(
    fn, name=None, dtype=None, **kwargs
)

參數

fn 要包裝的度量函數，帶有簽名 fn(y_true, y_pred, **kwargs) 。
name (可選)指標實例的字符串名稱。
dtype (可選)度量結果的數據類型。
**kwargs 要傳遞給 fn 的關鍵字參數。

您可以使用此類從函數中快速構建平均度量。該函數需要有簽名fn(y_true, y_pred)並返回一個per-sample損失數組。 MeanMetricWrapper.result() 將返回迄今為止看到的所有樣本的平均度量值。

例如：

def accuracy(y_true, y_pred):
  return tf.cast(tf.math.equal(y_true, y_pred), tf.float32)

accuracy_metric = tf.keras.metrics.MeanMetricWrapper(fn=accuracy)

keras_model.compile(..., metrics=accuracy_metric)

相關用法

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