a979a9419b90dfaaf50c04b4806748dbcfc0b379,tensorflow_ranking/python/losses_impl.py,ClickEMLoss,compute_unreduced_loss,#ClickEMLoss#,980

Before Change

        labels=exam_latent_prob, logits=exam_logits) * self._exam_loss_weight
    losses += tf.compat.v1.nn.sigmoid_cross_entropy_with_logits(
        labels=rel_latent_prob, logits=rel_logits) * self._rel_loss_weight
    return losses, tf.ones_like(losses)


class SigmoidCrossEntropyLoss(_PointwiseLoss):
  Implements sigmoid cross entropy loss.

After Change

        labels=exam_latent_prob, logits=exam_logits) * self._exam_loss_weight
    losses += tf.compat.v1.nn.sigmoid_cross_entropy_with_logits(
        labels=rel_latent_prob, logits=rel_logits) * self._rel_loss_weight
    return losses, tf.cast(mask, dtype=tf.float32)


class SigmoidCrossEntropyLoss(_PointwiseLoss):
  Implements sigmoid cross entropy loss.

Instances