tf.keras.Loss

Базовый класс функции потерь.

Основные псевдонимы

tf.keras.losses.Loss

Псевдонимы для совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.keras.Loss(
    name=None, reduction='sum_over_batch_size', dtype=None
)

Должны быть реализованы в подклассах:

call(): содержит логику вычисления функции потерь с использованием y_true, y_pred.

Пример реализации подкласса:

class MeanSquaredError(Loss):
    def call(self, y_true, y_pred):
        return ops.mean(ops.square(y_pred - y_true), axis=-1)

Методы

call(
    y_true, y_pred
)

@classmethod
from_config(
    config
)

get_config()

__call__(
    y_true, y_pred, sample_weight=None
)

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

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/Loss