Commit 5062d1e7 authored by Tiago Peixoto's avatar Tiago Peixoto

inference: keep track of entropy range in mcmc_equilibrate() when force_niter is not None

parent 08dfc58c
......@@ -135,6 +135,8 @@ def mcmc_equilibrate(state, wait=1000, nbreaks=2, max_niter=numpy.inf,
total_nattempts += nattempts
if force_niter is not None:
max_S = max(S, max_S)
min_S = min(S, min_S)
if niter >= force_niter:
break
else:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment