about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/BNT/inference/online/@jtree_2TBN_inf_engine/marginal_nodes.m
blob: e7ee2ae90083fdc8d26b4b008e8d26829865f091 (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
27
28
29
30
31
32
function marginal = marginal_nodes(engine, b, nodes, t, add_ev, is_fam)
% function marginal = marginal_nodes(engine, b, nodes, t, add_ev, is_fam) (jtree_2TBN)

if nargin < 6, is_fam = 0; end
ss = engine.slice_size;

if ~is_fam & (t > 1) & all(nodes<=ss)
  nodes = nodes + ss;
end

if t==1
  c = clq_containing_nodes(engine.jtree_engine1, nodes, is_fam);
else
  c = clq_containing_nodes(engine.jtree_engine, nodes, is_fam);
end
if c == -1
  error(['no clique contains ' nodes])
end
bigpot = b.clpot{c};
pot = marginalize_pot(bigpot, nodes, engine.maximize);
marginal = pot_to_marginal(pot);

% we convert the domain to the unrolled numbering system
% so that add_ev_to_dmarginal (maybe called in update_ess) extracts the right evidence.
if t > 1
  marginal.domain = nodes+(t-2)*engine.slice_size;
end
assert(~add_ev);