DecisionTreeFactor cardinalities getter

2023-06-28 18:25:53 -04:00 · 2023-06-28 18:25:53 -04:00 · c06861e7da
parent 2994c55b51
commit c06861e7da
1 changed files with 2 additions and 0 deletions
--- a/gtsam/discrete/DecisionTreeFactor.h
+++ b/gtsam/discrete/DecisionTreeFactor.h
@ -119,6 +119,8 @@ namespace gtsam {

    static double safe_div(const double& a, const double& b);

+    std::map<Key, size_t> cardinalities() const { return cardinalities_; }
+
    size_t cardinality(Key j) const { return cardinalities_.at(j); }

    /// divide by factor f (safely)