f9b98760f445fc0219cfc9c4cada2b5f9d35ab1b,basic/model.py,Model,_build_loss,#Model#,219

Before Change

        if config.wy:
            if config.na:
                na = tf.reshape(self.na, [-1, 1])
                concat_y = tf.concat(1, [na, tf.reshape(self.wy, [-1, M * JX])])
                losses = tf.nn.softmax_cross_entropy_with_logits(self.concat_logits, tf.cast(concat_y, "float"))
            else:
                losses = tf.nn.softmax_cross_entropy_with_logits(

After Change

            losses = tf.nn.sigmoid_cross_entropy_with_logits(
                tf.reshape(self.logits2, [-1, M, JX]), tf.cast(self.wy, "float"))  // [N, M, JX]
            num_pos = tf.reduce_sum(tf.cast(self.wy, "float"))
            num_neg = tf.reduce_sum(tf.cast(self.x_mask, "float")) - num_pos
            damp_ratio = num_pos / num_neg
            dampened_losses = losses * (
                (tf.cast(self.x_mask, "float") - tf.cast(self.wy, "float")) * damp_ratio + tf.cast(self.wy, "float"))
            new_losses = tf.reduce_sum(dampened_losses, [1, 2])
            ce_loss = tf.reduce_mean(loss_mask * new_losses)
            
            if config.na:
                na = tf.reshape(self.na, [-1, 1])

Instances