pymc.step_methods.hmc.HamiltonianMC.setup_chain

HamiltonianMC.setup_chain(rng, tune, draws)

Prepare the step method for sampling one chain.

Called once per chain, right before that chain starts sampling. The same instance may be reused across chains, so implementations must reset any per-chain state instead of accumulating it.

Parameters:

rngRandomGenerator

Random generator for this chain.

tuneint

Number of tuning iterations. Zero if the chain does not tune.

drawsint

Number of iterations after tuning. The chain runs tune + draws iterations in total.