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

add mixture to bayesnet and fix double assert bug

release/4.3a0
Varun Agrawal 2022-12-29 08:40:46 +05:30
parent 03baf8f75e
commit f4420f2c8d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python/gtsam/tests/test_HybridBayesNet.py
View File

@ -48,7 +48,7 @@ class TestHybridBayesNet(GtsamTestCase):
# Create hybrid Bayes net. # Create hybrid Bayes net.
bayesNet = HybridBayesNet() bayesNet = HybridBayesNet()
bayesNet.addGaussian(gc) bayesNet.addGaussian(gc)
# bayesNet.addMixture(gm) bayesNet.addMixture(gm)
bayesNet.addDiscrete(Asia, "99/1") bayesNet.addDiscrete(Asia, "99/1")
# Create values at which to evaluate. # Create values at which to evaluate.
@ -59,8 +59,8 @@ class TestHybridBayesNet(GtsamTestCase):
conditionalProbability = gc.evaluate(values.continuous()) conditionalProbability = gc.evaluate(values.continuous())
mixtureProbability = conditional0.evaluate(values.continuous()) mixtureProbability = conditional0.evaluate(values.continuous())
assert self.assertAlmostEqual(conditionalProbability * self.assertAlmostEqual(conditionalProbability * mixtureProbability *
mixtureProbability * 0.99, 0.99,
bayesNet.evaluate(values), bayesNet.evaluate(values),
places=5) places=5)