Beispiel #1
0
 def test_import_export(self):
     # write BN
     xmlbif = XMLBIF(self.bn, "Test Net")
     xmlbif.write("test_out.xmlbif")
     # read BN
     bn2 = XMLBIF.read("test_out.xmlbif")
     for node1 in self.bn.get_nodes():
         name_found = False
         cpd_equal = False
         value_range_equal = False
         str_equal = False
         pos_equal = False
         for node2 in bn2.get_nodes():
             print(node2.name)
             print(node2.get_cpd())
             # Test node names
             print(node2.name)
             if node1.name == node2.name:
                 name_found = True
                 cpd_equal = node1.get_cpd() == node2.get_cpd()
                 value_range_equal = node1.get_value_range() == node2.get_value_range()
                 str_equal = str(node1) == str(node2)
                 pos_equal = node1.pos == node2.pos
         self.assertTrue(name_found)
         self.assertTrue(cpd_equal)
         self.assertTrue(value_range_equal)
         self.assertTrue(str_equal)
         self.assertTrue(pos_equal)
     # remove file
     os.remove("test_out.xmlbif")
Beispiel #2
0
 def test_import_export(self):
     # write BN
     xmlbif = XMLBIF(self.bn, "Test Net")
     xmlbif.write("test_out.xmlbif")
     # read BN
     bn2 = XMLBIF.read("test_out.xmlbif")
     for node1 in self.bn.get_nodes():
         name_found = False
         cpd_equal = False
         value_range_equal = False
         str_equal = False
         pos_equal = False
         for node2 in bn2.get_nodes():
             print(node2.name)
             print(node2.get_cpd())
             # Test node names
             print(node2.name)
             if node1.name == node2.name:
                 name_found = True
                 cpd_equal = node1.get_cpd() == node2.get_cpd()
                 value_range_equal = node1.get_value_range(
                 ) == node2.get_value_range()
                 str_equal = str(node1) == str(node2)
                 pos_equal = node1.pos == node2.pos
         self.assertTrue(name_found)
         self.assertTrue(cpd_equal)
         self.assertTrue(value_range_equal)
         self.assertTrue(str_equal)
         self.assertTrue(pos_equal)
     # remove file
     os.remove("test_out.xmlbif")
# -*- coding: utf-8 -*-
import random

import numpy

from primo.networks import BayesianNetwork
from primo.networks import DynamicBayesianNetwork
from primo.networks import TwoTBN
from primo.nodes import DiscreteNode
import primo.inference.particlefilter as pf
from primo.io import XMLBIF

# Construct a DynmaicBayesianNetwork
dbn = DynamicBayesianNetwork()
B0 = BayesianNetwork()
twoTBN = TwoTBN(XMLBIF.read("Robot_Localization.xmlbif"))

# Configure TwoTBN
x0 = twoTBN.get_node("x0")
x = twoTBN.get_node("x")
door = twoTBN.get_node("door")
twoTBN.set_initial_node(x0.name, x.name)

# Configure initial distribution
x0_init = DiscreteNode(
    x0.name, ["p0", "p1", "p2", "p3", "p4", "p5", "p6", "p7", "p8", "p9"])
B0.add_node(x0_init)
cpt_x0_init = numpy.array([.1, .1, .1, .1, .1, .1, .1, .1, .1, .1])
x0_init.set_probability_table(cpt_x0_init, [x0_init])

dbn.B0 = B0