minor cleanup
parent
da5d3a242b
commit
812bf52c11
|
|
@ -14,7 +14,7 @@
|
||||||
* @brief Hybrid Bayes Tree, the result of eliminating a
|
* @brief Hybrid Bayes Tree, the result of eliminating a
|
||||||
* HybridJunctionTree
|
* HybridJunctionTree
|
||||||
* @date Mar 11, 2022
|
* @date Mar 11, 2022
|
||||||
* @author Fan Jiang
|
* @author Fan Jiang, Varun Agrawal
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <gtsam/base/treeTraversal-inst.h>
|
#include <gtsam/base/treeTraversal-inst.h>
|
||||||
|
|
@ -166,8 +166,6 @@ void HybridBayesTree::prune(const size_t maxNrLeaves) {
|
||||||
|
|
||||||
DecisionTreeFactor prunedDecisionTree = decisionTree->prune(maxNrLeaves);
|
DecisionTreeFactor prunedDecisionTree = decisionTree->prune(maxNrLeaves);
|
||||||
decisionTree->root_ = prunedDecisionTree.root_;
|
decisionTree->root_ = prunedDecisionTree.root_;
|
||||||
// this->print();
|
|
||||||
// decisionTree->print("", DefaultKeyFormatter);
|
|
||||||
|
|
||||||
/// Helper struct for pruning the hybrid bayes tree.
|
/// Helper struct for pruning the hybrid bayes tree.
|
||||||
struct HybridPrunerData {
|
struct HybridPrunerData {
|
||||||
|
|
|
||||||
|
|
@ -92,7 +92,6 @@ GaussianMixtureFactor::Sum sumFrontals(
|
||||||
if (auto cgmf = boost::dynamic_pointer_cast<GaussianMixtureFactor>(f)) {
|
if (auto cgmf = boost::dynamic_pointer_cast<GaussianMixtureFactor>(f)) {
|
||||||
sum = cgmf->add(sum);
|
sum = cgmf->add(sum);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (auto gm = boost::dynamic_pointer_cast<HybridConditional>(f)) {
|
if (auto gm = boost::dynamic_pointer_cast<HybridConditional>(f)) {
|
||||||
sum = gm->asMixture()->add(sum);
|
sum = gm->asMixture()->add(sum);
|
||||||
}
|
}
|
||||||
|
|
@ -189,7 +188,7 @@ hybridElimination(const HybridGaussianFactorGraph &factors,
|
||||||
DiscreteKeys discreteSeparator(discreteSeparatorSet.begin(),
|
DiscreteKeys discreteSeparator(discreteSeparatorSet.begin(),
|
||||||
discreteSeparatorSet.end());
|
discreteSeparatorSet.end());
|
||||||
|
|
||||||
// sum out frontals, this is the factor on the separator
|
// sum out frontals, this is the factor 𝜏 on the separator
|
||||||
GaussianMixtureFactor::Sum sum = sumFrontals(factors);
|
GaussianMixtureFactor::Sum sum = sumFrontals(factors);
|
||||||
|
|
||||||
// If a tree leaf contains nullptr,
|
// If a tree leaf contains nullptr,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue