From c06861e7da13e8fa67c89415797773a90611b316 Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Wed, 28 Jun 2023 18:25:53 -0400 Subject: [PATCH] DecisionTreeFactor cardinalities getter --- gtsam/discrete/DecisionTreeFactor.h | 2 ++ 1 file changed, 2 insertions(+) diff --git a/gtsam/discrete/DecisionTreeFactor.h b/gtsam/discrete/DecisionTreeFactor.h index 95054bcdb..0cfda6b7d 100644 --- 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 cardinalities() const { return cardinalities_; } + size_t cardinality(Key j) const { return cardinalities_.at(j); } /// divide by factor f (safely)