estimator.h

Go to the documentation of this file.

#ifndef GUM_ESTIMATOR_H
#define GUM_ESTIMATOR_H

#include <agrum/BN/IBayesNet.h>
#include <agrum/BN/inference/loopyBeliefPropagation.h>
#include <agrum/core/hashTable.h>
#include <vector>

namespace gum {

  template < typename GUM_SCALAR >
  class Estimator {
    public:
    Estimator();

    explicit Estimator(const IBayesNet< GUM_SCALAR >* bn);

    /* Destructor */
    ~Estimator();

    void setFromBN(const IBayesNet< GUM_SCALAR >* bn, const NodeSet& hardEvidence);

    void setFromLBP(LoopyBeliefPropagation< GUM_SCALAR >* lbp,
                    const NodeSet&                        hardEvidence,
                    GUM_SCALAR                            virtualLBPSize);

    GUM_SCALAR confidence();


    void update(Instantiation I, GUM_SCALAR w);


    const Potential< GUM_SCALAR >& posterior(const DiscreteVariable& var);


    void clear();

    private:
    HashTable< std::string, std::vector< GUM_SCALAR > > _estimator;

    GUM_SCALAR _wtotal;

    Size _ntotal;

    const IBayesNet< GUM_SCALAR >* _bn;


    GUM_SCALAR EV(std::string name, Idx val);


    GUM_SCALAR variance(std::string name, Idx val);   // variance corrigée

    private:

    HashTable< std::string, Potential< GUM_SCALAR >* > __target_posteriors;
  };

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

}   // namespace gum

#include <agrum/BN/inference/tools/estimator_tpl.h>
#endif