ea60242913a0ce53f4e033fee3be1f55e2ceb7b0,basenji/metrics.py,R2,update_state,#R2#Any#Any#Any#,249

Before Change

    y_true = tf.cast(y_true, "float32")
    y_pred = tf.cast(y_pred, "float32")

    true_sum = tf.reduce_sum(y_true, axis=[0,1])
    self._true_sum.assign_add(true_sum)

    true_sumsq = tf.reduce_sum(tf.math.square(y_true), axis=[0,1])
    self._true_sumsq.assign_add(true_sumsq)

After Change

    y_true = tf.cast(y_true, "float32")
    y_pred = tf.cast(y_pred, "float32")

    if len(y_true.shape) == 2:
      reduce_axes = 0
    else:
      reduce_axes = [0,1]

Instances