about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/BNT/inference/static/@jtree_inf_engine/Old/distribute_evidence.m
blob: f8d78be4e66f30c187c218796dfc5afaca7745a0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
function engine = distribute_evidence(engine, root)

if isempty(engine.preorder{root})
  % this is the first time we have distributed from this root
  % memoize the order
  [jtree, preorder, postorder] = mk_rooted_tree(engine.jtree, root);
  preorder_children = cell(1,length(preorder));
  for n=preorder
    preorder_children{n} = children(jtree, n);
  end
  engine.preorder{root} = preorder;
  engine.preorder_children{root} = preorder_children;
else
  preorder = engine.preorder{root};
  preorder_children = engine.preorder_children{root};
end


% distribute from root (node to children)
for n=preorder(:)'
  for c=preorder_children{n}(:)'
    engine.clpot{c} = divide_by_pot(engine.clpot{c}, engine.seppot{n,c}); 
    engine.seppot{n,c} = marginalize_pot(engine.clpot{n}, engine.separator{n,c}, engine.maximize);
    engine.clpot{c} = multiply_by_pot(engine.clpot{c}, engine.seppot{n,c});
  end
end