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')