diff options
| author | shivesh <s.p.mandalia@qmul.ac.uk> | 2018-11-08 15:02:27 -0600 |
|---|---|---|
| committer | shivesh <s.p.mandalia@qmul.ac.uk> | 2018-11-08 15:02:27 -0600 |
| commit | 845c55c269a59620bbf8a6c0d8adab575e1185dc (patch) | |
| tree | a8175935ba96947abc0ab678ce454bc938e3fd15 /utils/mcmc.py | |
| parent | 4f358ee507c06c1ccbce472817f4eeac8da5cfa0 (diff) | |
| download | GolemFlavor-845c55c269a59620bbf8a6c0d8adab575e1185dc.tar.gz GolemFlavor-845c55c269a59620bbf8a6c0d8adab575e1185dc.zip | |
Thu 8 Nov 15:02:27 CST 2018
Diffstat (limited to 'utils/mcmc.py')
| -rw-r--r-- | utils/mcmc.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/utils/mcmc.py b/utils/mcmc.py index fa044ea..e5bd8da 100644 --- a/utils/mcmc.py +++ b/utils/mcmc.py @@ -20,7 +20,7 @@ from utils.enums import MCMCSeedType from utils.misc import enum_parse, make_dir, parse_bool -def mcmc(p0, ln_prob, ndim, nwalkers, burnin, nsteps, threads=1): +def mcmc(p0, ln_prob, ndim, nwalkers, burnin, nsteps, args, threads=1): """Run the MCMC.""" sampler = emcee.EnsembleSampler( nwalkers, ndim, ln_prob, threads=threads @@ -31,6 +31,7 @@ def mcmc(p0, ln_prob, ndim, nwalkers, burnin, nsteps, threads=1): pos, prob, state = result sampler.reset() print "Finished burn-in" + args.burnin = False print "Running" for _ in tqdm.tqdm(sampler.sample(pos, iterations=nsteps), total=nsteps): |