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

計算 y_true 和 y_pred 之間的均方根誤差度量。

繼承自：Mean、Metric、Layer、Module

用法

tf.keras.metrics.RootMeanSquaredError(
    name='root_mean_squared_error', dtype=None
)

單機使用：

m = tf.keras.metrics.RootMeanSquaredError()
m.update_state([[0, 1], [0, 0]], [[1, 1], [0, 0]])
m.result().numpy()
0.5

m.reset_state()
m.update_state([[0, 1], [0, 0]], [[1, 1], [0, 0]],
               sample_weight=[1, 0])
m.result().numpy()
0.70710677

compile() API 的用法：

model.compile(
    optimizer='sgd',
    loss='mse',
    metrics=[tf.keras.metrics.RootMeanSquaredError()])