tf.keras.metrics.MeanMetricWrapper

Оборачивает функцию метрики без состояния метрикой Mean.

Наследуется от: Mean, Metric

tf.keras.metrics.MeanMetricWrapper(
    fn, name=None, dtype=None, **kwargs
)

Этот класс позволяет быстро создать метрику среднего значения на основе функции. Функция должна иметь сигнатуру fn(y_true, y_pred) и возвращать массив потерь для каждого образца. MeanMetricWrapper.result() возвращает среднее значение метрики по всем образцам, обработанным к текущему моменту.

Например:

def mse(y_true, y_pred):
    return (y_true - y_pred) ** 2

mse_metric = MeanMetricWrapper(fn=mse)

Методы

add_variable

Посмотреть исходный код

add_variable(
    shape, initializer, dtype=None, aggregation='sum', name=None
)

add_weight

Посмотреть исходный код

add_weight(
    shape=(), initializer=None, dtype=None, name=None
)

from_config

Посмотреть исходный код

@classmethod
from_config(
    config
)

get_config

Посмотреть исходный код

get_config()

Возвращает сериализуемую конфигурацию метрики.

reset_state

Посмотреть исходный код

reset_state()

Сбрасывает все переменные состояния метрики.

Эта функция вызывается между эпохами или шагами, когда метрика вычисляется в ходе обучения.

result

Посмотреть исходный код

result()

Вычисляет текущее значение метрики.

stateless_reset_state

Посмотреть исходный код

stateless_reset_state()

stateless_result

Посмотреть исходный код

stateless_result(
    metric_variables
)

stateless_update_state

Посмотреть исходный код

stateless_update_state(
    metric_variables, *args, **kwargs
)

update_state

Посмотреть исходный код

update_state(
    y_true, y_pred, sample_weight=None
)

Накапливает статистику метрики.

__call__

Посмотреть исходный код

__call__(
    *args, **kwargs
)

Вызывает объект как функцию.