5e329a5e3e9453ae0800ab438d2fa9f991f35a00,pymc3/examples/LKJ_correlation.py,,run,#Any#,49

Before Change

        start = pm.find_MAP()
        step = pm.NUTS(scaling=start)
        trace = pm.sample(n, step=step, start=start)
    return trace

if __name__ == "__main__":
    run()

After Change

    with model:
        trace = pm.sample(n)
    pm.traceplot(trace, varnames=["mu", "r"],
                 lines={"mu": mu_r, "r": corr_r[np.triu_indices(n_var, k=1)]})

if __name__ == "__main__":
    run()

Instances