This repository has been archived on 2023-08-20. You can view files and clone it, but cannot push or open issues or pull requests.
yap-6.3/packages/CLPBN/horus/VarElimSolver.h

47 lines
831 B
C
Raw Normal View History

2012-05-23 14:56:01 +01:00
#ifndef HORUS_VARELIMSOLVER_H
#define HORUS_VARELIMSOLVER_H
#include "unordered_map"
#include "Solver.h"
#include "FactorGraph.h"
#include "Horus.h"
using namespace std;
class VarElimSolver : public Solver
{
public:
VarElimSolver (const FactorGraph& fg) : Solver (fg) { }
~VarElimSolver (void);
Params solveQuery (VarIds);
void printSolverFlags (void) const;
private:
void createFactorList (void);
void absorveEvidence (void);
void findEliminationOrder (const VarIds&);
void processFactorList (const VarIds&);
void eliminate (VarId);
void printActiveFactors (void);
Factors factorList_;
VarIds elimOrder_;
unsigned largestFactorSize_;
unsigned totalFactorSize_;
2012-05-24 22:55:20 +01:00
unordered_map<VarId, vector<size_t>> varFactors_;
2012-05-23 14:56:01 +01:00
};
#endif // HORUS_VARELIMSOLVER_H