stepsize_covar_adapter.hpp

Go to the documentation of this file.

#ifndef STAN__MCMC__STEPSIZE__COVAR__ADAPTER__BETA
#define STAN__MCMC__STEPSIZE__COVAR__ADAPTER__BETA

#include <stan/mcmc/base_adapter.hpp>
#include <stan/mcmc/stepsize_adaptation.hpp>
#include <stan/mcmc/covar_adaptation.hpp>

namespace stan {
  
  namespace mcmc {
    
    class stepsize_covar_adapter: public base_adapter {
      
    public:
      
      stepsize_covar_adapter(int n): covar_adaptation_(n)
      {};
      
      stepsize_adaptation& get_stepsize_adaptation() {
        return stepsize_adaptation_;
      }
      
      covar_adaptation& get_covar_adaptation() {
        return covar_adaptation_;
      }
      
      void set_window_params(unsigned int num_warmup,
                             unsigned int init_buffer,
                             unsigned int term_buffer,
                             unsigned int base_window,
                             std::ostream* e = 0) {
        covar_adaptation_.set_window_params(num_warmup,
                                            init_buffer,
                                            term_buffer,
                                            base_window,
                                            e);
      }
      
    protected:
      
      stepsize_adaptation stepsize_adaptation_;
      covar_adaptation covar_adaptation_;
      
    };
    
  } // mcmc
  
} // stan

#endif