6bf3f51cd0a312da842157665663c2dad9983248,tensorflow_ranking/python/losses_impl.py,ClickEMLoss,_compute_latent_prob,#ClickEMLoss#,914

Before Change

              tf.compat.v1.where(is_clicked, tf.ones_like(prob), prob))
          for prob in [exam_prob_given_non_clicks, rel_prob_given_non_clicks]
      ]
      return exam_prob_given_non_clicks, rel_prob_given_non_clicks

  def compute_unreduced_loss(self, labels, logits):
    Computes the loss for each element.

After Change

      exam_prob_posterior = tf.compat.v1.where(
          is_clicked, tf.ones_like(exam_logits_posterior),
          tf.sigmoid(exam_logits_posterior))
      rel_prob_posterior = tf.compat.v1.where(
          is_clicked, tf.ones_like(rel_logits_posterior),
          tf.sigmoid(rel_logits_posterior))
      return tf.stop_gradient(exam_prob_posterior), tf.stop_gradient(
          rel_prob_posterior)

  def compute_unreduced_loss(self, labels, logits):

Instances