12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add print method to DiscreteMarginals

release/4.3a0
Varun Agrawal 2025-05-13 22:49:51 -04:00
parent 9f4ab83cee
commit 27fc6a7a4e
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
gtsam/discrete/DiscreteMarginals.cpp
View File

@ -55,4 +55,12 @@ Vector DiscreteMarginals::marginalProbabilities(const DiscreteKey& key) const {
return vResult; return vResult;
} }
/* ************************************************************************* */
void DiscreteMarginals::print(
const std::string& s = "",
const KeyFormatter formatter = DefaultKeyFormatter) const {
std::cout << (s.empty() ? "Discrete Marginals of:" : s + " ") << std::endl;
bayesTree_.print("", formatter);
}
} /* namespace gtsam */ } /* namespace gtsam */

4
gtsam/discrete/DiscreteMarginals.h
View File

@ -52,6 +52,10 @@ class DiscreteMarginals {
* @return Vector of marginal probabilities * @return Vector of marginal probabilities
*/ */
Vector marginalProbabilities(const DiscreteKey& key) const; Vector marginalProbabilities(const DiscreteKey& key) const;
/// Print details
void print(const std::string& s = "",
const KeyFormatter formatter = DefaultKeyFormatter) const;
}; };
} /* namespace gtsam */ } /* namespace gtsam */