about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/BNT/inference/dynamic/@ff_inf_engine/marginal_nodes.m
blob: f65a3bece17937d14ee7f0f305cfa88bc0173057 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
function marginal = marginal_nodes(engine, nodes, t)
% MARGINAL_NODES Compute the marginal on the specified query nodes (ff)
% marginal = marginal_nodes(engine, i, t)
% returns Pr(X(i,t) | Y(1:T)), where X(i,t) is the i'th node in the t'th slice.
% If enter_evidence used filtering instead of smoothing, this will return  Pr(X(i,t) | Y(1:t)).

if nargin < 3, t = 1; end
assert(length(nodes)==1);
i = nodes(end);
if myismember(i, engine.hnodes)
  marginal = pot_to_marginal(engine.marginals{i,t});
else
  marginal = pot_to_marginal(dpot(i, 1, 1)); % observed
end

bnet = bnet_from_engine(engine);
ss = length(bnet.intra);
% we convert the domain to the unrolled numbering system
% so that update_ess extracts the right evidence.
marginal.domain = nodes+(t-1)*ss;