Oblicza stratę Hubera między etykietami i prognozami.
Dla każdej wartości x w error = labels - predictions :
loss = 0.5 * x^2 if |x| <= d
loss = 0.5 * d^2 + d * (|x| - d) if |x| > d
gdzie d jest deltą.
Samodzielne użycie:
Operand<TFloat32> labels =
tf.constant(new float[][] { {0.f, 1.f}, {0.f, 0.f} });
Operand<TFloat32> predictions =
tf.constant(new float[][] { {0.6f, 0.4f}, {0.4f, 0.6f} });
Huber huberLoss = new Huber(tf);
Operand<TFloat32> result = huberLoss.call(labels, predictions);
// produces 0.155
Wywołanie z wagą próbki:
Operand<TFloat32> sampleWeight = tf.constant(new float[] {1.f, 0.f});
Operand<TFloat32> result = huberLoss.call(labels, predictions, sampleWeight);
// produces 0.09f
Używanie typu redukcji SUM :
Huber huberLoss = new Huber(tf, Reduction.SUM);
Operand<TFloat32> result = huberLoss.call(labels, predictions);
// produces 0.32f
Używanie typu redukcji NONE :
Huber huberLoss = new Huber(tf, Reduction.NONE);
Operand<TFloat32> result = huberLoss.call(labels, predictions);
// produces [0.18f, 0.13f]
Zobacz także
Stałe
| platforma | DELTA_DEFAULT |
Dziedziczone pola
Konstruktorzy publiczni
Huber (Ops tf) Tworzy stratę Hubera, używając getSimpleName() jako nazwy straty, DELTA_DEFAULT jako delty i redukcji straty wynoszącej REDUCTION_DEFAULT | |
Huber (Ops tf, nazwa ciągu) Tworzy stratę Hubera, używając DELTA_DEFAULT jako delty i redukcji straty wynoszącej REDUCTION_DEFAULT | |
Huber (Ops tf, redukcja redukcji ) Tworzy stratę Hubera, używając getSimpleName() jako nazwy straty i DELTA_DEFAULT jako delty | |
Huber (Ops tf, nazwa ciągu, redukcja redukcji ) Tworzy stratę Hubera, używając DELTA_DEFAULT jako delty | |
Metody publiczne
| <T rozszerza numer T > Operand <T> |
Metody dziedziczone
Stałe
publiczny statyczny końcowy float DELTA_DEFAULT
Konstruktorzy publiczni
publiczny Huber (Ops tf)
Tworzy stratę Hubera, używając getSimpleName() jako nazwy straty, DELTA_DEFAULT jako delty i redukcji straty wynoszącej REDUCTION_DEFAULT
Parametry
| tf | operacji TensorFlow |
|---|
publiczny Huber (Ops tf, nazwa ciągu)
Tworzy stratę Hubera, używając DELTA_DEFAULT jako delty i redukcji straty wynoszącej REDUCTION_DEFAULT
Parametry
| tf | operacji TensorFlow |
|---|---|
| nazwa | nazwa straty, jeśli ma wartość null, używana jest getSimpleName() . |
publiczny Huber (Ops tf, redukcja redukcji )
Tworzy stratę Hubera, używając getSimpleName() jako nazwy straty i DELTA_DEFAULT jako delty
Parametry
| tf | operacji TensorFlow |
|---|---|
| zmniejszenie | Rodzaj obniżki stosowanej w przypadku straty. |
publiczny Huber (Ops tf, nazwa ciągu, redukcja redukcji )
Tworzy stratę Hubera, używając DELTA_DEFAULT jako delty
Parametry
| tf | operacji TensorFlow |
|---|---|
| nazwa | nazwa straty, jeśli ma wartość null, używana jest getSimpleName() . |
| zmniejszenie | Rodzaj obniżki stosowanej w przypadku straty. |
publiczny Huber (Ops tf, nazwa ciągu, delta zmiennoprzecinkowa, redukcja redukcji )
Tworzy stratę Hubera
Parametry
| tf | operacji TensorFlow |
|---|---|
| nazwa | nazwa straty, jeśli ma wartość null, używana jest getSimpleName() . |
| delta | punkt, w którym funkcja straty Hubera zmienia się z kwadratowej na liniową. |
| zmniejszenie | Rodzaj obniżki stosowanej w przypadku straty. |
Metody publiczne
publiczne wywołanie argumentu <T> ( operand <? rozszerza TNumber > etykiety, przewidywania argumentu <T>, argumentu <T> próbkiWagi)
Generuje operand, który oblicza stratę.
Parametry
| etykiety | wartości prawdy lub etykiety |
|---|---|
| prognozy | przewidywania |
| próbkiWagi | Opcjonalne próbkiWeights działają jako współczynnik straty. Jeśli podany jest skalar, strata jest po prostu skalowana według podanej wartości. Jeśli SampleWeights jest tensorem rozmiaru [batch_size], wówczas całkowita strata dla każdej próbki w partii jest przeskalowana przez odpowiedni element wektora SampleWeights. Jeśli kształt SampleWeights wynosi [batch_size, d0, .. dN-1] (lub może być nadawany do tego kształtu), wówczas każdy element przewidywanej straty jest skalowany przez odpowiednią wartość SampleWeights. (Uwaga do dN-1: wszystkie funkcje straty zmniejszają się o 1 wymiar, zwykle oś=-1.) |
Powroty
- strata