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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。