237993ad641d386bf60f8414458c41a21c0a3c5b,pymc/tests/test_diagnostics.py,,test_effective_n,#Any#Any#,43

Before Change

def test_effective_n(k=3, n=1000):
    Unit test for effective sample size
    
    with nm.model:
        // Run sampler
        ptrace = sample(n, step, start, njobs=k,
                        random_seed=42)
        
    n_eff = effective_n(ptrace)
    
    assert np.isclose(n_eff, k*n, 2).all()

After Change

    
    model = Model()
    with model:
        x = Normal("x", 0, 1., shape=5)

        // start sampling at the MAP
        start = find_MAP()

Instances