wolffd@0
|
1 function marginal = marginal_family(engine, i, t)
|
wolffd@0
|
2 % MARGINAL_FAMILY Compute the marginal on the specified family (ff)
|
wolffd@0
|
3 % marginal = marginal_family(engine, i, t)
|
wolffd@0
|
4
|
wolffd@0
|
5 if nargin < 3, t = 1; end
|
wolffd@0
|
6
|
wolffd@0
|
7 % The method is similar to the following HMM equation:
|
wolffd@0
|
8 % xi(i,j,t) = normalise( alpha(i,t) * transmat(i,j) * obsmat(j,t+1) * beta(j,t+1) )
|
wolffd@0
|
9 % where xi(i,j,t) = Pr(Q(t)=i, Q(t+1)=j | y(1:T))
|
wolffd@0
|
10
|
wolffd@0
|
11 bnet = bnet_from_engine(engine);
|
wolffd@0
|
12
|
wolffd@0
|
13 if myismember(i, engine.onodes)
|
wolffd@0
|
14 ps = parents(bnet.dag, i);
|
wolffd@0
|
15 p = ps(1);
|
wolffd@0
|
16 marginal = pot_to_marginal(engine.marginals{p,t});
|
wolffd@0
|
17 marginal.domain = [p i];
|
wolffd@0
|
18 return;
|
wolffd@0
|
19 end
|
wolffd@0
|
20
|
wolffd@0
|
21 if t==1
|
wolffd@0
|
22 marginal = pot_to_marginal(engine.marginals{i,t});
|
wolffd@0
|
23 return;
|
wolffd@0
|
24 end
|
wolffd@0
|
25
|
wolffd@0
|
26 bnet = bnet_from_engine(engine);
|
wolffd@0
|
27 ss = length(bnet.intra);
|
wolffd@0
|
28 pot = engine.CPDpot{i,t};
|
wolffd@0
|
29 c = engine.obschild(i);
|
wolffd@0
|
30 pot = multiply_by_pot(pot, engine.CPDpot{c,t});
|
wolffd@0
|
31 pot = multiply_by_pot(pot, engine.back{i,t});
|
wolffd@0
|
32 ps = parents(bnet.dag, i+ss);
|
wolffd@0
|
33 for p=ps(:)'
|
wolffd@0
|
34 pot = multiply_by_pot(pot, engine.fwd{p,t-1});
|
wolffd@0
|
35 end
|
wolffd@0
|
36 marginal = pot_to_marginal(normalize_pot(pot));
|
wolffd@0
|
37
|
wolffd@0
|
38
|