Esempio n. 1
0
 def test_chow_liu1_F(self):
     bn = chow_liu(self.data)
     self.assertDictEqual(
         bn.F, {
             0: {
                 'cpt': [],
                 'parents': [],
                 'values': [1, 2]
             },
             1: {
                 'cpt': [],
                 'parents': [0],
                 'values': [1, 2]
             },
             2: {
                 'cpt': [],
                 'parents': [0],
                 'values': [1, 2]
             }
         })
Esempio n. 2
0
	def test_chow_liu1_F(self):
		bn = chow_liu(self.data)
		self.assertDictEqual(bn.F,
			{0: {'cpt': [], 'parents': [], 'values': [1,2]},
			 1: {'cpt': [], 'parents': [0], 'values': [1,2]},
			 2: {'cpt': [], 'parents': [0], 'values': [1,2]}})
Esempio n. 3
0
	def test_chow_liu1_E(self):
		bn = chow_liu(self.data)
		self.assertDictEqual(bn.E,
			{0:[1,2],1:[],2:[]})
Esempio n. 4
0
	def test_chow_liu1_V(self):
		bn = chow_liu(self.data)
		self.assertListEqual(bn.V,
			[0,1,2])
Esempio n. 5
0
 def test_chow_liu1_E(self):
     bn = chow_liu(self.data)
     self.assertDictEqual(bn.E, {0: [1, 2], 1: [], 2: []})
Esempio n. 6
0
 def test_chow_liu1_V(self):
     bn = chow_liu(self.data)
     self.assertListEqual(bn.V, [0, 1, 2])