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

additional Signature based constructor for DecisionTreeFactor

release/4.3a0
Varun Agrawal 2025-01-03 13:18:24 -05:00
parent 05d8030af4
commit 834288f974
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gtsam/discrete/DecisionTreeFactor.h
View File

@ -18,6 +18,7 @@
#pragma once #pragma once
#include <gtsam/discrete/Signature.h>
#include <gtsam/discrete/AlgebraicDecisionTree.h> #include <gtsam/discrete/AlgebraicDecisionTree.h>
#include <gtsam/discrete/DiscreteFactor.h> #include <gtsam/discrete/DiscreteFactor.h>
#include <gtsam/discrete/DiscreteKey.h> #include <gtsam/discrete/DiscreteKey.h>
@ -116,6 +117,10 @@ namespace gtsam {
DecisionTreeFactor(const DiscreteKey& key, const std::vector<double>& row) DecisionTreeFactor(const DiscreteKey& key, const std::vector<double>& row)
: DecisionTreeFactor(DiscreteKeys{key}, row) {} : DecisionTreeFactor(DiscreteKeys{key}, row) {}
/// Construct from Signature
DecisionTreeFactor(const Signature& signature)
: DecisionTreeFactor(signature.discreteKeys(), signature.cpt()) {}
/** Construct from a DiscreteConditional type */ /** Construct from a DiscreteConditional type */
explicit DecisionTreeFactor(const DiscreteConditional& c); explicit DecisionTreeFactor(const DiscreteConditional& c);