Struct nuts_rs::SamplerArgs

pub struct SamplerArgs {
    pub num_tune: u64,
    pub num_draws: u64,
    pub maxdepth: u64,
    pub store_gradient: bool,
    pub store_unconstrained: bool,
    pub max_energy_error: f64,
    pub store_divergences: bool,
    pub mass_matrix_adapt: GradDiagOptions,
}

Settings for the NUTS sampler

Fields

num_tune: u64

The number of tuning steps, where we fit the step size and mass matrix.

num_draws: u64

The number of draws after tuning

maxdepth: u64

The maximum tree depth during sampling. The number of leapfrog steps is smaller than 2 ^ maxdepth.

store_gradient: bool

Store the gradient in the SampleStats

store_unconstrained: bool

Store each unconstrained parameter vector in the sampler stats

max_energy_error: f64

If the energy error is larger than this threshold we treat the leapfrog step as a divergence.

store_divergences: bool

Store detailed information about each divergence in the sampler stats

mass_matrix_adapt: GradDiagOptions

Settings for mass matrix adaptation.

Trait Implementations

impl Clone for SamplerArgs

fn clone(&self) -> SamplerArgs

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for SamplerArgs

fn default() -> Self

Returns the “default value” for a type.

impl Copy for SamplerArgs