GibbsBNdistance.h

Go to the documentation of this file.

#ifndef GUM_GIBBS_KL2_H
#define GUM_GIBBS_KL2_H

#include <agrum/BN/algorithms/divergence/BNdistance.h>
#include <agrum/BN/inference/tools/gibbsOperator.h>
#include <agrum/core/approximations/approximationScheme.h>

#include <agrum/core/signal/signaler.h>

namespace gum {

  template < typename GUM_SCALAR >
  class GibbsBNdistance
      : public BNdistance< GUM_SCALAR >
      , public ApproximationScheme
      , public GibbsOperator< GUM_SCALAR > {
    public:
    /* no default constructor */

    GibbsBNdistance(const IBayesNet< GUM_SCALAR >& P,
                    const IBayesNet< GUM_SCALAR >& Q);

    explicit GibbsBNdistance(const BNdistance< GUM_SCALAR >& kl);

    ~GibbsBNdistance();

    void setBurnIn(Size b);

    Size burnIn() const;

    protected:
    void _computeKL() final;

    using BNdistance< GUM_SCALAR >::_p;
    using BNdistance< GUM_SCALAR >::_q;
    using BNdistance< GUM_SCALAR >::_hellinger;
    using BNdistance< GUM_SCALAR >::_bhattacharya;
    using BNdistance< GUM_SCALAR >::_jsd;

    using BNdistance< GUM_SCALAR >::_klPQ;
    using BNdistance< GUM_SCALAR >::_klQP;

    using BNdistance< GUM_SCALAR >::_errorPQ;
    using BNdistance< GUM_SCALAR >::_errorQP;
  };


#ifndef GUM_NO_EXTERN_TEMPLATE_CLASS
  extern template class GibbsBNdistance< double >;
#endif

}   // namespace gum

#include <agrum/BN/algorithms/divergence/GibbsKL_tpl.h>

#endif