9439f5215fde61ee40910b0320ff4dd8d28571e2,fancyimpute/bayesian_ridge_regression.py,BayesianRidgeRegression,predict_dist,#BayesianRidgeRegression#,84

Before Change

        else:
            X_ones = X
        mus = self.predict(X)
        covar = self.sigma_squared_estimate * self.inverse_covariance
        sigmas_squared = self.sigma_squared_estimate + np.sum(
            np.dot(X_ones, covar) * X_ones, axis=1)
        return mus, sigmas_squared

After Change

        rescaled_X = dot(X_ones, self.covar)
        rescaled_X *= X_ones
        sigmas_squared = rescaled_X.sum(axis=1)
        sigmas_squared += self.sigma_squared_estimate
        return mus, sigmas_squared

Instances