tf.keras.losses.MeanSquaredError

Вычисляет среднее значение квадратов ошибок между метками и прогнозами.

tf.keras.losses.MeanSquaredError(
    reduction=losses_utils.ReductionV2.AUTO, name='mean_squared_error'
)

loss = square(y_true - y_pred)

Использование в автономном режиме:

y_true = [[0., 1.], [0., 0.]]
y_pred = [[1., 1.], [1., 0.]]
# Using 'auto'/'sum_over_batch_size' reduction type.
mse = tf.keras.losses.MeanSquaredError()
mse(y_true, y_pred).numpy()
0.5

# Calling with 'sample_weight'.
mse(y_true, y_pred, sample_weight=[0.7, 0.3]).numpy()
0.25

# Using 'sum' reduction type.
mse = tf.keras.losses.MeanSquaredError(
    reduction=tf.keras.losses.Reduction.SUM)
mse(y_true, y_pred).numpy()
1.0

# Using 'none' reduction type.
mse = tf.keras.losses.MeanSquaredError(
    reduction=tf.keras.losses.Reduction.NONE)
mse(y_true, y_pred).numpy()
array([0.5, 0.5], dtype=float32)

Использование с API compile():

model.compile(optimizer='sgd', loss=tf.keras.losses.MeanSquaredError())

Методы

from_config

Просмотр исходного кода

@classmethod
from_config(
    config
)

Инициализирует Loss из его конфигурации (результат get_config()).

get_config

Просмотр исходного кода

get_config()

Возвращает словарь конфигурации для экземпляра Loss.

__call__

Просмотр исходного кода

__call__(
    y_true, y_pred, sample_weight=None
)

Вызывает экземпляр Loss.