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

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

继承自：Metric、Layer、Module

用法

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

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

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

如果 sample_weight 是 None ，则权重默认为 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')