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

emplace then prune

release/4.3a0
Varun Agrawal 2025-01-04 17:22:58 -05:00
parent 07a68296d5
commit bcc52becfb
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
gtsam/hybrid/HybridBayesNet.cpp
View File

@ -61,13 +61,15 @@ HybridBayesNet HybridBayesNet::prune(size_t maxNrLeaves) const {
} }
} }
// Prune the joint. NOTE: imperative and, again, possibly quite expensive. // Create the result starting with the pruned joint.
DiscreteConditional pruned(joint);
pruned.prune(maxNrLeaves);
// Create a the result starting with the pruned joint.
HybridBayesNet result; HybridBayesNet result;
result.push_back(std::make_shared<DiscreteConditional>(pruned)); 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
// we can prune HybridGaussianConditionals.
DiscreteConditional pruned = *result.back()->asDiscrete();
/* To prune, we visitWith every leaf in the HybridGaussianConditional. /* To prune, we visitWith every leaf in the HybridGaussianConditional.
* For each leaf, using the assignment we can check the discrete decision tree * For each leaf, using the assignment we can check the discrete decision tree