exactBNdistance.h

Go to the documentation of this file.

#ifndef GUM_BRUTE_FORCE_KL_H
#define GUM_BRUTE_FORCE_KL_H

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

namespace gum {

  template < typename GUM_SCALAR >
  class ExactBNdistance : public BNdistance< GUM_SCALAR > {
    public:
    ExactBNdistance(const IBayesNet< GUM_SCALAR >& P,
                    const IBayesNet< GUM_SCALAR >& Q);

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

    virtual ~ExactBNdistance();

    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 ExactBNdistance< double >;
#endif

}   // namespace gum

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

#endif   // GUM_BRUTE_FORCE_KL_H