littlemcmc.QuadPotentialDiagAdapt

class littlemcmc.QuadPotentialDiagAdapt(n, initial_mean, initial_diag=None, initial_weight=0, adaptation_window=101, adaptation_window_multiplier=1, dtype=None)

Adapt a diagonal mass matrix from the sample variances.

__init__(n, initial_mean, initial_diag=None, initial_weight=0, adaptation_window=101, adaptation_window_multiplier=1, dtype=None)

Set up a diagonal mass matrix.

Methods

__init__(n, initial_mean[, initial_diag, …])	Set up a diagonal mass matrix.
energy(x[, velocity])	Compute kinetic energy at a position in parameter space.
raise_ok(vmap)	Check if the mass matrix is ok, and raise ValueError if not.
random()	Draw random value from QuadPotential.
reset()	Reset quadpotential adaptation routine.
update(sample, grad, tune)	Inform the potential about a new sample during tuning.
velocity(x[, out])	Compute the current velocity at a position in parameter space.
velocity_energy(x, v_out)	Compute velocity and return kinetic energy at a position in parameter space.