MeanMetricWrapper

clase pública MeanMetricWrapper
Subclases directas conocidas

Una clase que une una función de pérdida sin estado con la métrica Mean mediante una reducción de WEIGHTED_MEAN .

La función de pérdida calcula la pérdida entre las labels y las predictions y luego pasa esta pérdida a la métrica Mean para calcular la media ponderada de la pérdida durante muchas iteraciones o épocas.

Constantes heredadas

Métodos públicos

LossMetric <T>
getLoss ()
Obtiene la función de pérdida.
Lista < Op >
updateStateList ( operando <? extiende TNumber > etiquetas, operando <? extiende TNumber > predicciones, operando <? extiende TNumber > sampleWeights)
Crea operaciones que actualizan el estado de la métrica media, llamando a la función de pérdida y pasando la pérdida a la métrica Media para calcular la media ponderada de la pérdida en muchas iteraciones.

Métodos heredados

Métodos públicos

public LossMetric <T> getLoss ()

Obtiene la función de pérdida.

Devoluciones
  • la función de pérdida.

Public List < Op > updateStateList ( operando <? extiende TNumber > etiquetas, operando <? extiende TNumber > predicciones, operando <? extiende TNumber > sampleWeights)

Crea operaciones que actualizan el estado de la métrica media, llamando a la función de pérdida y pasando la pérdida a la métrica Media para calcular la media ponderada de la pérdida en muchas iteraciones.

Parámetros
etiquetas los valores de verdad o etiquetas
predicciones las predicciones
sampleWeights SampleWeights opcional actúa como un coeficiente para la pérdida. Si se proporciona un escalar, entonces la pérdida simplemente se escala por el valor dado. Si sampleWeights es un tensor de tamaño [batch_size], entonces la pérdida total para cada muestra del lote se reescala por el elemento correspondiente en el vector sampleWeights. Si la forma de sampleWeights es [batch_size, d0, .. dN-1] (o puede transmitirse con esta forma), entonces cada elemento de pérdida de las predicciones se escala por el valor correspondiente de sampleWeights. (Nota sobre dN-1: todas las funciones de pérdida se reducen en 1 dimensión, generalmente eje = -1).
Devoluciones
  • a Lista de operaciones de control que actualiza las variables de estado Media.