b6f83ab5a36233c57b19aa38f28d58a80b19c3ba,ludwig/modules/metric_modules.py,R2Score,__init__,#R2Score#Any#,47

Before Change

    def __init__(self, name="r2_score", **kwargs):
        super(R2Score, self).__init__(name=name)
        self._reset_states()

    def _reset_states(self):
        self.sum_y = 0.0
        self.sum_y_squared = 0.0

After Change

            "sum_y_hat_squared", initializer="zeros",
            dtype=tf.float32
        )
        self.sum_y_hat = self.add_weight(
            "sum_y_y_hat", initializer="zeros",
            dtype=tf.float32
        )
        self.sum_y_y_hat = self.add_weight(
            "sum_y_y_hat", initializer="zeros",
            dtype=tf.float32
        )
        self.N = self.add_weight(
            "N", initializer="zeros",
            dtype=tf.float32
        )

Instances