48afd8f2c4ee6f746769642ffac135f3caab3a65,arviz/stats/stats.py,,r2_score,#Any#Any#Any#,513

Before Change

    
    y_pred = np.atleast_2d(y_pred).mean(0)

    y_pred_ = []
    for _ in range(1000):
        idx = np.random.randint(0, len(y_true), size=len(y_true))
        y_pred_.append(_r_square(y_true, y_pred, idx))

After Change

    
    if y_pred.ndim == 1:
        var_y_est = np.var(y_pred)
        var_e = np.var(y_true - y_pred)
    else:
        var_y_est = np.var(y_pred.mean(0))
        var_e = np.var(y_true - y_pred, 0)

    r2 = var_y_est / (var_y_est + var_e)

Instances