NUTSKernel.start_epoch()

NUTSKernel.start_epoch()#

NUTSKernel.start_epoch(prng_key, kernel_state, model_state, epoch)[source]#

Resets the state of the dual averaging algorithm.

Return type:

NUTSKernelState