Struct nuts_rs::SamplerArgs

pub struct SamplerArgs {
    pub num_tune: u64,
    pub maxdepth: u64,
    pub store_gradient: bool,
    pub max_energy_error: f64,
    pub step_size_adapt: DualAverageSettings,
    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.

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

max_energy_error: f64

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

step_size_adapt: DualAverageSettings

Settings for step size adaptation.

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