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

discreteKeys method

release/4.3a0
Frank Dellaert 2022-01-22 11:06:52 -05:00
parent b875914f93
commit eb4309d264
2 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
gtsam/discrete/DecisionTreeFactor.cpp
View File

@ -67,7 +67,7 @@ namespace gtsam {
void DecisionTreeFactor::print(const string& s,
const KeyFormatter& formatter) const {
cout << s;
ADT::print("Potentials:",formatter);
ADT::print("", formatter);
}
/* ************************************************************************* */
@ -163,6 +163,18 @@ namespace gtsam {
return result;
}
/* ************************************************************************* */
DiscreteKeys DecisionTreeFactor::discreteKeys() const {
DiscreteKeys result;
for (auto&& key : keys()) {
DiscreteKey dkey(key, cardinality(key));
if (std::find(result.begin(), result.end(), dkey) == result.end()) {
result.push_back(dkey);
}
}
return result;
}
/* ************************************************************************* */
static std::string valueFormatter(const double& v) {
return (boost::format("%4.2g") % v).str();

3
gtsam/discrete/DecisionTreeFactor.h
View File

@ -183,6 +183,9 @@ namespace gtsam {
/// Enumerate all values into a map from values to double.
std::vector<std::pair<DiscreteValues, double>> enumerate() const;
/// Return all the discrete keys associated with this factor.
DiscreteKeys discreteKeys() const;
/// @}
/// @name Wrapper support
/// @{