Mercurial > hg > camir-aes2014
comparison toolboxes/FullBNT-1.0.7/bnt/inference/static/@belprop_inf_engine/private/tree_protocol.m @ 0:e9a9cd732c1e tip
first hg version after svn
author | wolffd |
---|---|
date | Tue, 10 Feb 2015 15:05:51 +0000 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:e9a9cd732c1e |
---|---|
1 function bel = tree_protocol(engine, evidence, pot_type, local_kernel, msg) | |
2 | |
3 bnet = bnet_from_engine(engine); | |
4 ns = bnet.node_sizes; | |
5 onodes = find(~isemptycell(evidence)); | |
6 | |
7 ndoms = length(engine.gdl.doms); | |
8 prod_of_msg = cell(1, ndoms); | |
9 bel = cell(1, ndoms); | |
10 | |
11 % collect to root (node to parents) | |
12 for n=engine.postorder | |
13 % absorb msgs from children | |
14 prod_of_msg{n} = mk_initial_pot(pot_type, engine.gdl.doms{n}, ns, bnet.cnodes, onodes); | |
15 for c=children(engine.tree, n) | |
16 prod_of_msg{n} = multiply_by_pot(prod_of_msg{n}, msg{c,n}); | |
17 end | |
18 % send msg to parents | |
19 for p=parents(engine.tree, n) | |
20 if iter==1 | |
21 temp = prod_of_msg{n}; | |
22 else | |
23 temp = divide_by_pot(prod_of_msg{n}, old_msg{p,n}); | |
24 end | |
25 temp = multiply_by_pot(temp, local_kernel{n}); | |
26 temp2 = marginalize_pot(temp, engine.gdl.sepset{n,p}, engine.maximize); | |
27 %fprintf('%d sends %d\n', n, p); | |
28 msg{n,p} = normalize_pot(temp2); | |
29 end | |
30 end | |
31 | |
32 % distribute from root (node to children) | |
33 for n=engine.preorder | |
34 % absorb from parents | |
35 %prod_of_msg{n} = mk_initial_pot(pot_type, doms{n}, ns, cnodes, onodes); | |
36 for p=parents(engine.tree, n) | |
37 prod_of_msg{n} = multiply_by_pot(prod_of_msg{n}, msg{p,n}); | |
38 end | |
39 bel{n} = normalize_pot(multiply_pots(prod_of_msg{n}, local_kernel{n})); | |
40 % send msg to children | |
41 for c=children(engine.tree, n) | |
42 temp = divide_by_pot(prod_of_msg{n}, msg{c,n}); | |
43 temp = multiply_by_pot(temp, local_kernel{n}); | |
44 temp2 = marginalize_pot(temp, engine.gdl.sepset{n,c}, engine.maximize); | |
45 %fprintf('%d sends %d\n', n, c); | |
46 msg{n,c} = normalize_pot(temp2); | |
47 end | |
48 end |