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

prune and then add to BayesNet

release/4.3a0
Varun Agrawal 2025-01-24 18:04:33 -05:00
parent 26642f1ba0
commit 2644c4baa9
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gtsam/hybrid/HybridBayesNet.cpp
View File

@ -58,11 +58,12 @@ HybridBayesNet HybridBayesNet::prune(size_t maxNrLeaves,
joint = joint * (*conditional); joint = joint * (*conditional);
} }
// Prune the joint. NOTE: imperative and, again, possibly quite expensive.
joint.prune(maxNrLeaves);
// Create the result starting with the pruned joint. // Create the result starting with the pruned joint.
HybridBayesNet result; HybridBayesNet result;
result.emplace_shared<DiscreteConditional>(joint); result.emplace_shared<DiscreteConditional>(joint);
// Prune the joint. NOTE: imperative and, again, possibly quite expensive.
result.back()->asDiscrete()->prune(maxNrLeaves);
// Get pruned discrete probabilities so // Get pruned discrete probabilities so
// we can prune HybridGaussianConditionals. // we can prune HybridGaussianConditionals.