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

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

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