Stan  2.10.0
probability, sampling & optimization
adapt_diag_e_static_uniform.hpp
Go to the documentation of this file.
1 #ifndef STAN_MCMC_HMC_STATIC_UNIFORM_ADAPT_DIAG_E_STATIC_UNIFORM_HPP
2 #define STAN_MCMC_HMC_STATIC_UNIFORM_ADAPT_DIAG_E_STATIC_UNIFORM_HPP
3 
6 
7 namespace stan {
8  namespace mcmc {
15  template <typename Model, class BaseRNG>
17  public diag_e_static_uniform<Model, BaseRNG>,
18  public stepsize_var_adapter {
19  public:
20  adapt_diag_e_static_uniform(const Model& model, BaseRNG& rng):
21  diag_e_static_uniform<Model, BaseRNG>(model, rng),
22  stepsize_var_adapter(model.num_params_r()) {}
23 
25 
26  sample
27  transition(sample& init_sample,
30  sample s
32  info_writer,
33  error_writer);
34 
35  if (this->adapt_flag_) {
37  s.accept_stat());
38 
39  bool update = this->var_adaptation_.learn_variance(this->z_.mInv,
40  this->z_.q);
41  if (update) {
42  this->init_stepsize(info_writer, error_writer);
43  this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_));
45  }
46  }
47  return s;
48  }
49 
53  }
54  };
55  } // mcmc
56 } // stan
57 
58 #endif
void complete_adaptation(double &epsilon)
adapt_diag_e_static_uniform(const Model &model, BaseRNG &rng)
double accept_stat() const
Definition: sample.hpp:41
Probability, optimization and sampling library.
void learn_stepsize(double &epsilon, double adapt_stat)
Hamiltonian Monte Carlo implementation that uniformly samples from trajectories with a static integra...
Hamiltonian Monte Carlo implementation that uniformly samples from trajectories with a static integra...
base_writer is an abstract base class defining the interface for Stan writer callbacks.
Definition: base_writer.hpp:20
virtual void disengage_adaptation()
sample transition(sample &init_sample, interface_callbacks::writer::base_writer &info_writer, interface_callbacks::writer::base_writer &error_writer)
bool learn_variance(Eigen::VectorXd &var, const Eigen::VectorXd &q)
void init_stepsize(interface_callbacks::writer::base_writer &info_writer, interface_callbacks::writer::base_writer &error_writer)
Definition: base_hmc.hpp:64
sample transition(sample &init_sample, interface_callbacks::writer::base_writer &info_writer, interface_callbacks::writer::base_writer &error_writer)

     [ Stan Home Page ] © 2011–2016, Stan Development Team.