def test_find_MAP():
print '-' * 80
G = MarkovModel()
G.add_nodes_from(['x1', 'x2', 'x3'])
G.add_edges_from([('x1', 'x2'), ('x1', 'x3')])
phi = [
DiscreteFactor(['x2', 'x1'],
cardinality=[2, 2],
values=np.array([[1.0 / 1, 1.0 / 2], [1.0 / 3,
1.0 / 4]])),
DiscreteFactor(['x3', 'x1'],
cardinality=[2, 2],
values=np.array([[1.0 / 1, 1.0 / 2], [1.0 / 3,
1.0 / 4]]))
]
# DiscreteFactor(['x1'], cardinality=[2],
# values=np.array([2,2]))]
G.add_factors(*phi)
print "nodes:", G.nodes()
bp = BeliefPropagation(G)
bp.max_calibrate()
# bp.calibrate()
clique_beliefs = bp.get_clique_beliefs()
print clique_beliefs
print clique_beliefs[('x1', 'x2')]
print clique_beliefs[('x1', 'x3')]
# print 'partition function should be', np.sum(clique_beliefs[('x1', 'x3')].values)
phi_query = bp._query(['x1', 'x2', 'x3'], operation='maximize')
# phi_query = bp._query(['x1', 'x2', 'x3'], operation='marginalize')
print phi_query
sleep(52)
def find_MAP_state(G):
'''
Inputs:
- G: MarkovModel
'''
bp = BeliefPropagation(G)
bp.max_calibrate()
clique_beliefs = bp.get_clique_beliefs()
phi_query = bp._query(G.nodes(), operation='maximize')
print phi_query
return phi_query
