wrap DiscreteMarginals class

release/4.3a0
Varun Agrawal 2025-05-13 22:50:30 -04:00
parent 27fc6a7a4e
commit 5333317079
1 changed files with 14 additions and 0 deletions

View File

@ -494,4 +494,18 @@ class DiscreteSearch {
std::vector<gtsam::DiscreteSearchSolution> run(size_t K = 1) const;
};
#include <gtsam/discrete/DiscreteMarginals.h>
class DiscreteMarginals {
DiscreteMarginals();
DiscreteMarginals(const gtsam::DiscreteFactorGraph& graph);
gtsam::DiscreteFactor* operator()(gtsam::Key variable) const;
gtsam::Vector marginalProbabilities(const gtsam::DiscreteKey& key) const;
void print(const std::string& s = "",
const gtsam::KeyFormatter& keyFormatter =
gtsam::DefaultKeyFormatter) const;
};
} // namespace gtsam