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
)

Вызов self как функции.