b6f83ab5a36233c57b19aa38f28d58a80b19c3ba,ludwig/modules/metric_modules.py,R2Score,update_state,#R2Score#Any#Any#,62

Before Change

        self._reset_states()

    def update_state(self, y, y_hat):
        self.sum_y += np.sum(y)
        self.sum_y_squared += np.sum(y ** 2)
        self.sum_y_hat += np.sum(y_hat)
        self.sum_y_hat_squared += np.sum(y_hat ** 2)
        self.sum_y_y_hat += np.sum(y * y_hat)

After Change

        )

    def update_state(self, y, y_hat):
        y = tf.cast(y, dtype=tf.float32)
        y_hat = tf.cast(y_hat, dtype=tf.float32)
        self.sum_y.assign_add(tf.reduce_sum(y))
        self.sum_y_squared.assign_add(tf.reduce_sum(y ** 2))
        self.sum_y_hat.assign_add(tf.reduce_sum(y_hat))
        self.sum_y_hat_squared.assign_add(tf.reduce_sum(y_hat ** 2))

Instances