61570406a54b612908b37f4028cd47df4a70781a,batchflow/models/metrics/regression.py,RegressionMetrics,r2_score,#RegressionMetrics#,109

Before Change

    def r2_score(self):
         r2_score 
        numerator = (self.predictions - self.targets) ** 2
        denominator = (self.targets - np.average(self.targets)) ** 2
        return np.mean(1 - (numerator / denominator))

    def explained_variance_ratio(self):

After Change

    def r2_score(self):
         r2_score 
        if self.weights is not None:
            weight = self.weights[:, np.newaxis]
        else:
            weight = 1
        numerator = (weight * (self.predictions - self.targets) ** 2).sum(axis=0)
        targets_avg = np.average(self.targets, axis=0, weights=self.weights)
        denominator = (weight * (self.targets - targets_avg) ** 2).sum(axis=0)
        return 1 - (numerator / denominator)

Instances