85cb4440b63de8744ddfc9a213f2cdad535d5224,tensorflow_ranking/python/losses_impl.py,SigmoidCrossEntropyLoss,compute_unreduced_loss,#SigmoidCrossEntropyLoss#Any#Any#Any#,1025

Before Change

        utils.is_label_valid(labels), logits, tf.zeros_like(logits))
    losses = tf.compat.v1.nn.sigmoid_cross_entropy_with_logits(
        labels=labels, logits=logits)
    return losses, tf.ones_like(losses)


class MeanSquaredLoss(_PointwiseLoss):
  Implements the means squared error loss.

After Change

  def compute_unreduced_loss(self, labels, logits, mask=None):
    See `_RankingLoss`.
    if mask is None:
      mask = utils.is_label_valid(labels)
    labels = tf.compat.v1.where(mask, labels, tf.zeros_like(labels))
    logits = tf.compat.v1.where(mask, logits, tf.zeros_like(logits))
    losses = tf.compat.v1.nn.sigmoid_cross_entropy_with_logits(
        labels=labels, logits=logits)
    return losses, tf.cast(mask, dtype=tf.float32)


class MeanSquaredLoss(_PointwiseLoss):
  Implements the means squared error loss.

Instances