GibbsSampling.h

Go to the documentation of this file.

#ifndef GUM_GIBBS_SAMPLING_H
#define GUM_GIBBS_SAMPLING_H

#include <agrum/BN/inference/tools/gibbsOperator.h>
#include <agrum/BN/inference/tools/samplingInference.h>


namespace gum {

  template < typename GUM_SCALAR >
  class GibbsSampling
      : public SamplingInference< GUM_SCALAR >
      , public GibbsOperator< GUM_SCALAR > {
    public:
    explicit GibbsSampling(const IBayesNet< GUM_SCALAR >* bn);

    ~GibbsSampling() override;

    void setBurnIn(Size b) { this->_burn_in = b; };

    Size burnIn() const { return this->_burn_in; };

    protected:
    Instantiation _burnIn() override;


    Instantiation _draw(GUM_SCALAR* w, Instantiation prev) override;


    Instantiation _monteCarloSample();
  };


#ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
  extern template class GibbsSampling< double >;
#endif
}   // namespace gum

#include <agrum/BN/inference/GibbsSampling_tpl.h>
#endif