Add support for more infernce keys.

Add support for export_libdai, export_uai, export_gv and print_fg.
Document these keys.
This commit is contained in:
Tiago Gomes
2013-01-08 17:01:03 +00:00
parent 4220069d90
commit da0514a779
6 changed files with 148 additions and 42 deletions

View File

@@ -32,6 +32,23 @@ main (int argc, const char* argv[])
FactorGraph fg;
readFactorGraph (fg, argv[idx]);
VarIds queryIds = readQueryAndEvidence (fg, argc, argv, idx + 1);
if (FactorGraph::exportToLibDai()) {
fg.exportToLibDai ("model.fg");
}
if (FactorGraph::exportToUai()) {
fg.exportToUai ("model.uai");
}
if (FactorGraph::exportGraphViz()) {
fg.exportToGraphViz ("model.dot");
}
if (FactorGraph::printFactorGraph()) {
fg.print();
}
if (Globals::verbosity > 0) {
cout << "factor graph contains " ;
cout << fg.nrVarNodes() << " variables and " ;
cout << fg.nrFacNodes() << " factors " << endl;
}
runSolver (fg, queryIds);
return 0;
}