From b343a8096544d8a7468969d773094d6a09fd21ca Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Wed, 1 Jan 2025 22:26:31 -0500 Subject: [PATCH] more helper methods in DiscreteTableConditional --- gtsam/discrete/DiscreteTableConditional.h | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/gtsam/discrete/DiscreteTableConditional.h b/gtsam/discrete/DiscreteTableConditional.h index a8f187d2c..b722015c6 100644 --- a/gtsam/discrete/DiscreteTableConditional.h +++ b/gtsam/discrete/DiscreteTableConditional.h @@ -189,6 +189,14 @@ class GTSAM_EXPORT DiscreteTableConditional : public DiscreteConditional { virtual DiscreteConditional::shared_ptr prune( size_t maxNrAssignments) const override; + /// Get a DecisionTreeFactor representation. + DecisionTreeFactor toDecisionTreeFactor() const override { + return table_.toDecisionTreeFactor(); + } + + /// Get the number of non-zero values. + size_t nrValues() const { return table_.sparseTable().nonZeros(); } + /// @} private: