当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.keras.metrics.Mean用法及代码示例


计算给定值的(加权)平均值。

继承自:MetricLayerModule

用法

tf.keras.metrics.Mean(
    name='mean', dtype=None
)

参数

  • name (可选)指标实例的字符串名称。
  • dtype (可选)度量结果的数据类型。

例如,如果值为 [1, 3, 5, 7],则平均值为 4。如果权重指定为 [1, 1, 0, 0],则平均值将为 2。

此指标创建两个变量 totalcount 用于计算 values 的平均值。该平均值最终以 mean 的形式返回,这是一个幂等运算,只需将 total 除以 count

如果 sample_weightNone ,则权重默认为 1。使用 0 的 sample_weight 来屏蔽值。

单机使用:

m = tf.keras.metrics.Mean()
m.update_state([1, 3, 5, 7])
m.result().numpy()
4.0
m.reset_state()
m.update_state([1, 3, 5, 7], sample_weight=[1, 1, 0, 0])
m.result().numpy()
2.0

compile() API 的用法:

model.add_metric(tf.keras.metrics.Mean(name='mean_1')(outputs))
model.compile(optimizer='sgd', loss='mse')

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.metrics.Mean。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。