47 lines
868 B
C++
47 lines
868 B
C++
#ifndef YAP_PACKAGES_CLPBN_HORUS_VARELIM_H_
|
|
#define YAP_PACKAGES_CLPBN_HORUS_VARELIM_H_
|
|
|
|
#include <vector>
|
|
#include <unordered_map>
|
|
|
|
#include "GroundSolver.h"
|
|
#include "FactorGraph.h"
|
|
#include "Horus.h"
|
|
|
|
|
|
namespace Horus {
|
|
|
|
class VarElim : public GroundSolver {
|
|
public:
|
|
VarElim (const FactorGraph& fg) : GroundSolver (fg) { }
|
|
|
|
~VarElim() { }
|
|
|
|
Params solveQuery (VarIds);
|
|
|
|
void printSolverFlags() const;
|
|
|
|
private:
|
|
void createFactorList();
|
|
|
|
void absorveEvidence();
|
|
|
|
Params processFactorList (const VarIds&);
|
|
|
|
void eliminate (VarId);
|
|
|
|
void printActiveFactors();
|
|
|
|
Factors factorList_;
|
|
unsigned largestFactorSize_;
|
|
unsigned totalFactorSize_;
|
|
std::unordered_map<VarId, std::vector<size_t>> varMap_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN (VarElim);
|
|
};
|
|
|
|
} // namespace Horus
|
|
|
|
#endif // YAP_PACKAGES_CLPBN_HORUS_VARELIM_H_
|
|
|