MeanMetricWrapper

klasa publiczna MeanMetricWrapper
Znane podklasy bezpośrednie

Klasa łącząca funkcję straty bezstanowej z metryką Mean przy użyciu redukcji WEIGHTED_MEAN .

Funkcja straty oblicza stratę między labels i predictions , a następnie przekazuje tę stratę do metryki Mean , aby obliczyć średnią ważoną straty w wielu iteracjach lub epokach

Dziedziczone stałe

Metody publiczne

Metryka strat <T>
spierniczaj ()
Pobiera funkcję straty.
Lista <Op>
updateStateList ( Operand <? rozszerza TNumber > etykiety, Operand <? rozszerza TNumber > przewidywania, Operand <? rozszerza TNumber > próbkiWagi)
Tworzy operacje, które aktualizują stan metryki średniej, wywołując funkcję straty i przekazując stratę do metryki Mean w celu obliczenia średniej ważonej straty w wielu iteracjach.

Metody dziedziczone

Metody publiczne

publiczny LossMetric <T> getLoss ()

Pobiera funkcję straty.

Zwroty
  • funkcja straty.

public List< Op > updateStateList ( Operand <? rozszerza TNumber > etykiety, Operand <? rozszerza TNumber > przewidywania, Operand <? rozszerza TNumber > próbkiWagi)

Tworzy operacje, które aktualizują stan metryki średniej, wywołując funkcję straty i przekazując stratę do metryki Mean w celu obliczenia średniej ważonej straty w wielu iteracjach.

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ć rozgłaszany do tego kształtu), wówczas każdy element predykcji straty jest skalowany przez odpowiednią wartość sampleWeights. (Uwaga do dN-1: wszystkie funkcje strat zmniejszają się o 1 wymiar, zwykle oś=-1.)
Zwroty
  • a Lista operacji kontrolnych aktualizujących zmienne stanu średniego.