configs.hpp

Go to the documentation of this file.

#pragma once
#include <cstddef>
#include <autoppl/mcmc/hmc/step_adapter.hpp>
#include <autoppl/mcmc/hmc/momentum_handler.hpp>
#include <autoppl/mcmc/sampler_tools.hpp>
#include <autoppl/mcmc/config_base.hpp>
 
namespace ppl {
 
template <class VarAdapterPolicy=diag_var>
struct NUTSConfig: ConfigBase
{
    using var_adapter_policy_t = VarAdapterPolicy;
 
    // configuration for sampling
    size_t max_depth = 10;
 
    // configuration for step-size adaptation
    StepConfig step_config;
 
    // configuration for variance adaptation
    VarConfig var_config;
};
 
template <class NUTSConfigType>
struct nuts_config_traits
{
    using var_adapter_policy_t = 
        typename NUTSConfigType::var_adapter_policy_t;
};
 
} // namespace ppl