46 lines
830 B
C++
46 lines
830 B
C++
#ifndef HORUS_LIFTEDBP_H
|
|
#define HORUS_LIFTEDBP_H
|
|
|
|
#include "LiftedSolver.h"
|
|
#include "ParfactorList.h"
|
|
|
|
class FactorGraph;
|
|
class WeightedBp;
|
|
|
|
class LiftedBp : public LiftedSolver
|
|
{
|
|
public:
|
|
LiftedBp (const ParfactorList& pfList);
|
|
|
|
~LiftedBp (void);
|
|
|
|
Params solveQuery (const Grounds&);
|
|
|
|
void printSolverFlags (void) const;
|
|
|
|
private:
|
|
void refineParfactors (void);
|
|
|
|
bool iterate (void);
|
|
|
|
vector<PrvGroup> getQueryGroups (const Grounds&);
|
|
|
|
void createFactorGraph (void);
|
|
|
|
vector<vector<unsigned>> getWeights (void) const;
|
|
|
|
unsigned rangeOfGround (const Ground&);
|
|
|
|
Params getJointByConditioning (const ParfactorList&, const Grounds&);
|
|
|
|
ParfactorList pfList_;
|
|
WeightedBp* solver_;
|
|
FactorGraph* fg_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN (LiftedBp);
|
|
|
|
};
|
|
|
|
#endif // HORUS_LIFTEDBP_H
|
|
|