0b64530ff2cc6390ea3e56c77ce081ed3a3d3c9a,tensorflow_ranking/python/losses.py,_ListMLELoss,compute_unreduced_loss,#_ListMLELoss#Any#Any#Any#,1113

Before Change

    weights = tf.squeeze(weights)

    // Shuffle labels and logits to add randomness to sort.
    shuffled_indices = utils.shuffle_valid_indices(is_label_valid, self._seed)
    shuffled_labels = tf.gather_nd(labels, shuffled_indices)
    shuffled_logits = tf.gather_nd(logits, shuffled_indices)

    // TODO: Remove the shuffling above and use
    // shuffle_ties=True.

After Change

        tf.reduce_min(labels, axis=1, keepdims=True) -
        1e-6 * tf.ones_like(labels))
    sorted_labels, sorted_logits = utils.sort_by_scores(
        scores, [labels, logits], shuffle_ties=True)

    raw_max = tf.reduce_max(input_tensor=sorted_logits, axis=1, keepdims=True)
    sorted_logits = sorted_logits - raw_max

Instances