drop suport for Pearl belief propagation

This commit is contained in:
Tiago Gomes
2012-04-05 19:34:37 +01:00
parent 0d23591058
commit abb0410d07
6 changed files with 6 additions and 1187 deletions

View File

@@ -3,18 +3,13 @@
#include <iostream>
#include <sstream>
#include "BayesNet.h"
#include "FactorGraph.h"
#include "VarElimSolver.h"
#include "BnBpSolver.h"
#include "FgBpSolver.h"
#include "CbpSolver.h"
#include "ElimGraph.h"
using namespace std;
void processArguments (BayesNet&, int, const char* []);
void processArguments (FactorGraph&, int, const char* []);
void runSolver (Solver*, const VarNodes&);
@@ -25,53 +20,24 @@ const string USAGE = "usage: \
int
main (int argc, const char* argv[])
{
VarIds vids1 = { 4, 1, 2, 3 } ;
VarIds vids2 = { 4, 5 } ;
VarIds vids3 = { 4, 6 } ;
VarIds vids4 = { 4, 7 } ;
// Factor f1 (vids1, {2,2,2,2},{0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,0.10,0.11,0.12,0.13,0.14,0.15,0.16});
// Factor f2 (vids2, {2,2},{0.1,0.2,0.3,0.4});
// Factor f3 (vids3, {2,2},{0.1,0.2,0.3,0.4});
// Factor f4 (vids4, {2,2},{0.1,0.2,0.3,0.4});
Factor* f1 = new Factor (vids1, {2,2,2,2},{0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,0.10,0.11,0.12,0.13,0.14,0.15,0.16});
Factor* f2 = new Factor (vids2, {2,2},{0.1,0.2,0.3,0.4});
Factor* f3 = new Factor (vids3, {2,2},{0.1,0.2,0.3,0.4});
Factor* f4 = new Factor (vids4, {2,2},{0.1,0.2,0.3,0.4});
Factor* f5 = new Factor (vids4, {2,2},{0.1,0.2,0.3,0.4});
vector<Factor*> fs = {f1,f2,f3,f4,f5};
//FactorGraph fg;
//fg.addFactor (f1);
//fg.addFactor (f2);
//fg.addFactor (f3);
//fg.addFactor (f4);
ElimGraph eg (fs);
eg.exportToGraphViz ("_eg.dot");
return 0;
if (!argv[1]) {
cerr << "error: no graphical model specified" << endl;
cerr << USAGE << endl;
exit (0);
}
const string& fileName = argv[1];
const string& extension = fileName.substr (fileName.find_last_of ('.') + 1);
if (extension == "xml") {
BayesNet bn;
bn.readFromBifFormat (argv[1]);
processArguments (bn, argc, argv);
} else if (extension == "uai") {
FactorGraph fg;
const string& extension = fileName.substr (
fileName.find_last_of ('.') + 1);
FactorGraph fg;
if (extension == "uai") {
fg.readFromUaiFormat (argv[1]);
processArguments (fg, argc, argv);
} else if (extension == "fg") {
FactorGraph fg;
fg.readFromLibDaiFormat (argv[1]);
processArguments (fg, argc, argv);
} else {
cerr << "error: the graphical model must be defined either " ;
cerr << "in a xml, uai or libDAI file" << endl;
cerr << "in a UAI or libDAI file" << endl;
exit (0);
}
return 0;
@@ -79,83 +45,6 @@ main (int argc, const char* argv[])
void
processArguments (BayesNet& bn, int argc, const char* argv[])
{
VarNodes queryVars;
for (int i = 2; i < argc; i++) {
const string& arg = argv[i];
if (arg.find ('=') == std::string::npos) {
BayesNode* queryVar = bn.getBayesNode (arg);
if (queryVar) {
queryVars.push_back (queryVar);
} else {
cerr << "error: there isn't a variable labeled of " ;
cerr << "`" << arg << "'" ;
cerr << endl;
exit (0);
}
} else {
size_t pos = arg.find ('=');
const string& label = arg.substr (0, pos);
const string& state = arg.substr (pos + 1);
if (label.empty()) {
cerr << "error: missing left argument" << endl;
cerr << USAGE << endl;
exit (0);
}
if (state.empty()) {
cerr << "error: missing right argument" << endl;
cerr << USAGE << endl;
exit (0);
}
BayesNode* node = bn.getBayesNode (label);
if (node) {
if (node->isValidState (state)) {
node->setEvidence (state);
} else {
cerr << "error: `" << state << "' " ;
cerr << "is not a valid state for " ;
cerr << "`" << node->label() << "'" ;
cerr << endl;
exit (0);
}
} else {
cerr << "error: there isn't a variable labeled of " ;
cerr << "`" << label << "'" ;
cerr << endl;
exit (0);
}
}
}
Solver* solver = 0;
FactorGraph* fg = 0;
switch (Globals::infAlgorithm) {
case InfAlgorithms::VE:
fg = new FactorGraph (bn);
solver = new VarElimSolver (*fg);
break;
case InfAlgorithms::BN_BP:
solver = new BnBpSolver (bn);
break;
case InfAlgorithms::FG_BP:
fg = new FactorGraph (bn);
solver = new FgBpSolver (*fg);
break;
case InfAlgorithms::CBP:
fg = new FactorGraph (bn);
solver = new CbpSolver (*fg);
break;
default:
assert (false);
}
runSolver (solver, queryVars);
delete fg;
}
void
processArguments (FactorGraph& fg, int argc, const char* argv[])
{
@@ -238,7 +127,6 @@ processArguments (FactorGraph& fg, int argc, const char* argv[])
case InfAlgorithms::VE:
solver = new VarElimSolver (fg);
break;
case InfAlgorithms::BN_BP:
case InfAlgorithms::FG_BP:
solver = new FgBpSolver (fg);
break;