Se usó la API de Cloud Translation para traducir esta página.
Switch to English

tf.keras.losses.MeanSquaredError

TensorFlow 1 versión Ver código fuente en GitHub

Calcula la media de los cuadrados de los errores entre las etiquetas y las predicciones.

Se utiliza en los cuadernos

Se utiliza en la guía Se utiliza en los tutoriales

loss = square(y_true - y_pred)

el uso autónomo:

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)

Uso con la compile() API:

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

reduction (Opcional) Tipo de tf.keras.losses.Reduction que aplica a las pérdidas. El valor por defecto es AUTO . AUTO indica que la opción de reducción será determinado por el contexto de uso. Para casi todos los casos el valor predeterminado es SUM_OVER_BATCH_SIZE . Cuando se utiliza con tf.distribute.Strategy , fuera de una función de formación tales como bucles tf.keras compile y fit , utilizando AUTO o SUM_OVER_BATCH_SIZE se producirá un error. Por favor, vea esta formación personalizada tutorial para más detalles.
name nombre opcional para la operación. Por defecto es 'mean_squared_error'.

Métodos

from_config

Ver fuente

Instancia un Loss de su configuración (salida de get_config() ).

args
config Salida de get_config() .

Devoluciones
Una Loss ejemplo.

get_config

Ver fuente

Devuelve el diccionario de configuración para una Loss ejemplo.

__call__

Ver fuente

Invoca la Loss ejemplo.

args
y_true valores de verdad de tierra. forma = [batch_size, d0, .. dN] , excepto las funciones de pérdida dispersos tales como crossentropy categórica escasa donde forma = [batch_size, d0, .. dN-1]
y_pred Los valores pronosticados. forma = [batch_size, d0, .. dN]
sample_weight Opcional sample_weight actúa como un coeficiente para la pérdida. Si se proporciona un escalar, entonces la pérdida es simplemente escalado por el valor dado. Si sample_weight es un tensor de tamaño [batch_size] , entonces la pérdida total para cada muestra de la tanda se reajustarán por el elemento correspondiente en el sample_weight vector. Si la forma de sample_weight es [batch_size, d0, .. dN-1] (o puede ser transmitido a esta forma), entonces cada elemento de pérdida de y_pred es escalado por el valor correspondiente de sample_weight . (Nota sobre dN-1 : todas las funciones de pérdida reducen por 1 dimensión, por lo general eje = -1).

Devoluciones
Flotador pérdida ponderada Tensor . Si reduction es NONE , este tiene la forma [batch_size, d0, .. dN-1] ; de lo contrario, es escalar. (Nota dN-1 porque todas las funciones de pérdida reducen por 1 dimensión, por lo general eje = -1).

aumentos
ValueError Si la forma de sample_weight no es válido.