90c7286c62845066565fa9af7f058d5a03179fb2,pymc3/tests/test_stats.py,TestStats,test_waic,#TestStats#,70

Before Change

        log_py = st.binom.logpmf(np.atleast_2d(x_obs).T, 5, trace["p"]).T

        lppd = np.sum(np.log(np.mean(np.exp(log_py), axis=0)))
        p_waic = np.sum(np.var(log_py, axis=0))
        actual = -2 * lppd +  * p_waic

        assert_almost_equal(calculated, actual, decimal=2)

After Change

        log_py = st.binom.logpmf(np.atleast_2d(x_obs).T, 5, trace["p"]).T

        lppd_i = np.log(np.mean(np.exp(log_py), axis=0))
        vars_lpd = np.var(log_py, axis=0)
        waic_i = - 2 * (lppd_i - vars_lpd)
        
        actual_waic_se = np.sqrt(len(waic_i) * np.var(waic_i))
        actual_waic = np.sum(waic_i)
        
        assert_almost_equal(calculated_waic, actual_waic, decimal=2)

Instances