85cb4440b63de8744ddfc9a213f2cdad535d5224,tensorflow_ranking/python/losses_impl.py,SigmoidCrossEntropyLoss,compute_unreduced_loss,#SigmoidCrossEntropyLoss#,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

    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