wolffd@0
|
1 function CPD = update_ess(CPD, fmarginal, evidence, ns, cnodes, hidden_bitv)
|
wolffd@0
|
2 % UPDATE_ESS Update the Expected Sufficient Statistics of a hhmm Q node.
|
wolffd@0
|
3 % function CPD = update_ess(CPD, fmarginal, evidence, ns, cnodes, idden_bitv)
|
wolffd@0
|
4 %
|
wolffd@0
|
5 % we assume if one of the Qps is observed, all of them are
|
wolffd@0
|
6 % We assume the F nodes are already hidden
|
wolffd@0
|
7
|
wolffd@0
|
8 % Figure out the node numbers associated with each parent
|
wolffd@0
|
9 dom = fmarginal.domain;
|
wolffd@0
|
10 self = dom(CPD.self_ndx);
|
wolffd@0
|
11 old_self = dom(CPD.old_self_ndx);
|
wolffd@0
|
12 %Fself = dom(CPD.Fself_ndx);
|
wolffd@0
|
13 %Fbelow = dom(CPD.Fbelow_ndx);
|
wolffd@0
|
14 Qps = dom(CPD.Qps_ndx);
|
wolffd@0
|
15
|
wolffd@0
|
16 Qsz = CPD.Qsz;
|
wolffd@0
|
17 Qpsz = CPD.Qpsz;
|
wolffd@0
|
18
|
wolffd@0
|
19
|
wolffd@0
|
20 % hor_counts(old_self, Qps, self),
|
wolffd@0
|
21 % fmarginal(old_self, Fbelow, Fself, Qps, self)
|
wolffd@0
|
22 % hor_counts(i,k,j) = fmarginal(i,2,1,k,j) % below has finished, self has not
|
wolffd@0
|
23 % ver_counts(i,k,j) = fmarginal(i,2,2,k,j) % below has finished, and so has self (reset)
|
wolffd@0
|
24 % Since any of i,j,k may be observed, we write
|
wolffd@0
|
25 % hor_counts(i_counts_ndx, kndx, jndx) = fmarginal(i_fmarg_ndx...)
|
wolffd@0
|
26 % where i_fmarg_ndx = 1 and i_counts_ndx = i if old_self is observed to have value i,
|
wolffd@0
|
27 % i_fmarg_ndx = 1:Qsz and i_counts_ndx = 1:Qsz if old_self is hidden, etc.
|
wolffd@0
|
28
|
wolffd@0
|
29
|
wolffd@0
|
30 if hidden_bitv(old_self)
|
wolffd@0
|
31 i_counts_ndx = 1:Qsz;
|
wolffd@0
|
32 i_fmarg_ndx = 1:Qsz;
|
wolffd@0
|
33 eff_oldQsz = Qsz;
|
wolffd@0
|
34 else
|
wolffd@0
|
35 i_counts_ndx = evidence{old_self};
|
wolffd@0
|
36 i_fmarg_ndx = 1;
|
wolffd@0
|
37 eff_oldQsz = 1;
|
wolffd@0
|
38 end
|
wolffd@0
|
39
|
wolffd@0
|
40 if all(hidden_bitv(Qps)) % we assume all are hidden or all are observed
|
wolffd@0
|
41 k_counts_ndx = 1:Qpsz;
|
wolffd@0
|
42 k_fmarg_ndx = 1:Qpsz;
|
wolffd@0
|
43 eff_Qpsz = Qpsz;
|
wolffd@0
|
44 else
|
wolffd@0
|
45 k_counts_ndx = subv2ind(Qpsz, cat(1, evidence{Qps}));
|
wolffd@0
|
46 k_fmarg_ndx = 1;
|
wolffd@0
|
47 eff_Qpsz = 1;
|
wolffd@0
|
48 end
|
wolffd@0
|
49
|
wolffd@0
|
50 if hidden_bitv(self)
|
wolffd@0
|
51 j_counts_ndx = 1:Qsz;
|
wolffd@0
|
52 j_fmarg_ndx = 1:Qsz;
|
wolffd@0
|
53 eff_Qsz = Qsz;
|
wolffd@0
|
54 else
|
wolffd@0
|
55 j_counts_ndx = evidence{self};
|
wolffd@0
|
56 j_fmarg_ndx = 1;
|
wolffd@0
|
57 eff_Qsz = 1;
|
wolffd@0
|
58 end
|
wolffd@0
|
59
|
wolffd@0
|
60 hor_counts = zeros(Qsz, Qpsz, Qsz);
|
wolffd@0
|
61 ver_counts = zeros(Qpsz, Qsz);
|
wolffd@0
|
62
|
wolffd@0
|
63 if ~isempty(CPD.Fbelow_ndx)
|
wolffd@0
|
64 if ~isempty(CPD.Fself_ndx) % general case
|
wolffd@0
|
65 fmarg.T = myreshape(fmarg.T, [eff_oldQsz 2 2 eff_Qpsz eff_Qsz]);
|
wolffd@0
|
66 hor_counts(i_counts_ndx, k_counts_ndx, j_counts_ndx) = ...
|
wolffd@0
|
67 fmarg.T(:, i_fmarg_ndx, 2, 1, k_fmarg_ndx, j_fmarg_ndx);
|
wolffd@0
|
68 ver_counts(k_counts_ndx, j_counts_ndx) = ... % sum over Fbelow and Qold
|
wolffd@0
|
69 sum(fmarg.T(:, 1, 2, k_fmarg_ndx, j_fmarg_ndx), 1) + ...
|
wolffd@0
|
70 sum(fmarg.T(:, 2, 2, k_fmarg_ndx, j_fmarg_ndx), 1);
|
wolffd@0
|
71 else % no F from self, hence no startprob
|
wolffd@0
|
72 fmarg.T = myreshape(fmarg.T, [eff_oldQsz 2 eff_Qpsz eff_Qsz]);
|
wolffd@0
|
73 hor_counts(i_counts_ndx, k_counts_ndx, j_counts_ndx) = ...
|
wolffd@0
|
74 fmarg.T(i_fmarg_ndx, 2, k_fmarg_ndx, j_fmarg_ndx);
|
wolffd@0
|
75 end
|
wolffd@0
|
76 else % no F signal from below
|
wolffd@0
|
77 if ~isempty(CPD.Fself_ndx) % self F
|
wolffd@0
|
78 fmarg.T = myreshape(fmarg.T, [eff_oldQsz 2 eff_Qpsz eff_Qsz]);
|
wolffd@0
|
79 hor_counts(i_counts_ndx, k_counts_ndx, j_counts_ndx) = ...
|
wolffd@0
|
80 fmarg.T(i_fmarg_ndx, 1, k_fmarg_ndx, j_fmarg_ndx);
|
wolffd@0
|
81 ver_counts(k_counts_ndx, j_counts_ndx) = ... % sum over Qold
|
wolffd@0
|
82 sum(fmarg.T(:, 2, k_fmarg_ndx, j_fmarg_ndx), 1);
|
wolffd@0
|
83 else % no F from self
|
wolffd@0
|
84 error('An hhmmQ node without any F parents is just a tabular_CPD')
|
wolffd@0
|
85 end
|
wolffd@0
|
86 end
|
wolffd@0
|
87
|
wolffd@0
|
88
|
wolffd@0
|
89 CPD.sub_CPD_trans = update_ess_simple(CPD.sub_CPD_trans, hor_counts);
|
wolffd@0
|
90
|
wolffd@0
|
91 if ~isempty(CPD.sub_CPD_start)
|
wolffd@0
|
92 CPD.sub_CPD_start = update_ess_simple(CPD.sub_CPD_start, ver_counts);
|
wolffd@0
|
93 end
|
wolffd@0
|
94
|
wolffd@0
|
95
|