tf.keras.losses.Loss

Базовый класс потерь.

tf.keras.losses.Loss(
    reduction=losses_utils.ReductionV2.AUTO, name=None
)

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

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

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

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

При использовании с tf.distribute.Strategy вне встроенных циклов обучения, таких как tf.keras compile и fit, используйте типы редукции 'SUM' или 'NONE', и явно уменьшайте потери в цикле обучения. Использование 'AUTO' или 'SUM_OVER_BATCH_SIZE' вызовет ошибку.

Дополнительные сведения см. в https://www.tensorflow.org/alpha/tutorials/distribute/training_loops.

Вы можете реализовать 'SUM_OVER_BATCH_SIZE', используя глобальный размер пакета, например:

with strategy.scope():
  loss_obj = tf.keras.losses.CategoricalCrossentropy(
      reduction=tf.keras.losses.Reduction.NONE)
  ....
  loss = (tf.reduce_sum(loss_obj(labels, predictions)) *
          (1. / global_batch_size))

Методы

call

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

@abc.abstractmethod
call(
    y_true, y_pred
)

Вызывает экземпляр Loss.

from_config

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

@classmethod
from_config(
    config
)

Создает экземпляр Loss из его конфигурации (выход get_config()).

get_config

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

get_config()

__call__

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

__call__(
    y_true, y_pred, sample_weight=None
)

Вызывает экземпляр Loss.