42 lines
1014 B
C++
42 lines
1014 B
C++
#ifndef HORUS_VARELIMSOLVER_H
|
|
#define HORUS_VARELIMSOLVER_H
|
|
|
|
#include "unordered_map"
|
|
|
|
#include "Solver.h"
|
|
#include "FactorGraph.h"
|
|
#include "BayesNet.h"
|
|
#include "Shared.h"
|
|
|
|
|
|
using namespace std;
|
|
|
|
|
|
class VarElimSolver : public Solver
|
|
{
|
|
public:
|
|
VarElimSolver (const BayesNet&);
|
|
VarElimSolver (const FactorGraph&);
|
|
~VarElimSolver (void);
|
|
void runSolver (void) { }
|
|
ParamSet getPosterioriOf (VarId);
|
|
ParamSet getJointDistributionOf (const VarIdSet&);
|
|
|
|
private:
|
|
void createFactorList (void);
|
|
void introduceEvidence (void);
|
|
void chooseEliminationOrder (const VarIdSet&);
|
|
void processFactorList (const VarIdSet&);
|
|
void eliminate (VarId);
|
|
void printActiveFactors (void);
|
|
|
|
const BayesNet* bayesNet_;
|
|
const FactorGraph* factorGraph_;
|
|
vector<Factor*> factorList_;
|
|
VarIdSet elimOrder_;
|
|
unordered_map<VarId, vector<unsigned>> varFactors_;
|
|
};
|
|
|
|
#endif // HORUS_VARELIMSOLVER_H
|
|
|