wolffd@0
|
1 % Make the HHMM in Figure 1 of the NIPS'01 paper
|
wolffd@0
|
2
|
wolffd@0
|
3 Qsize = [2 3 2];
|
wolffd@0
|
4 Qnodes = 1:3;
|
wolffd@0
|
5 D = 3;
|
wolffd@0
|
6 transprob = cell(1,D);
|
wolffd@0
|
7 termprob = cell(1,D);
|
wolffd@0
|
8 startprob = cell(1,D);
|
wolffd@0
|
9 clear A;
|
wolffd@0
|
10
|
wolffd@0
|
11 % transprob{d}(i,k,j), transprob{1}(i,j)
|
wolffd@0
|
12 % termprob{d}(k,j), termprob{1}(1,j)
|
wolffd@0
|
13 % startprob{d}(k,j), startprob{1}(1,j)
|
wolffd@0
|
14
|
wolffd@0
|
15
|
wolffd@0
|
16 % LEVEL 1
|
wolffd@0
|
17
|
wolffd@0
|
18 % 1 2 e
|
wolffd@0
|
19 A{1} = [0 0 1;
|
wolffd@0
|
20 0 0 1];
|
wolffd@0
|
21 [transprob{1}, termprob{1}] = remove_hhmm_end_state(A{1});
|
wolffd@0
|
22 startprob{1} = [0.5 0.5];
|
wolffd@0
|
23
|
wolffd@0
|
24 % LEVEL 2
|
wolffd@0
|
25 A{2} = zeros(Qsize(2), Qsize(1), Qsize(2)+1);
|
wolffd@0
|
26
|
wolffd@0
|
27 % 1 2 3 e
|
wolffd@0
|
28 A{2}(:,1,:) = [0 1 0 0 % Q1=1 => model below state 0
|
wolffd@0
|
29 0 0 1 0
|
wolffd@0
|
30 0 0 0 1];
|
wolffd@0
|
31
|
wolffd@0
|
32 % 1 2 3 e
|
wolffd@0
|
33 A{2}(:,2,:) = [0 1 0 0 % Q1=2 => model below state 1
|
wolffd@0
|
34 0 0 1 0
|
wolffd@0
|
35 0 0 0 1];
|
wolffd@0
|
36
|
wolffd@0
|
37 [transprob{2}, termprob{2}] = remove_hhmm_end_state(A{2});
|
wolffd@0
|
38
|
wolffd@0
|
39 % always enter level 2 in state 1
|
wolffd@0
|
40 startprob{2} = [1 0 0
|
wolffd@0
|
41 1 0 0];
|
wolffd@0
|
42
|
wolffd@0
|
43 % LEVEL 3
|
wolffd@0
|
44
|
wolffd@0
|
45 A{3} = zeros([Qsize(3) Qsize(2) Qsize(3)+1]);
|
wolffd@0
|
46 endstate = Qsize(3)+1;
|
wolffd@0
|
47 % Qt-1(3) Qt(2) Qt(3)
|
wolffd@0
|
48 % 1 2 e
|
wolffd@0
|
49 A{3}(1, 1, endstate) = 1.0; % Q2=1 => model below state 2/5
|
wolffd@0
|
50 A{3}(:, 2, :) = [0.0 1.0 0.0 % Q2=2 => model below state 3/6
|
wolffd@0
|
51 0.5 0.0 0.5];
|
wolffd@0
|
52 A{3}(1, 3, endstate) = 1.0; % Q2=3 => model below state 4/7
|
wolffd@0
|
53
|
wolffd@0
|
54 [transprob{3}, termprob{3}] = remove_hhmm_end_state(A{3});
|
wolffd@0
|
55
|
wolffd@0
|
56 startprob{3} = 'leftstart';
|
wolffd@0
|
57
|
wolffd@0
|
58
|
wolffd@0
|
59
|
wolffd@0
|
60 % OBS LEVEl
|
wolffd@0
|
61
|
wolffd@0
|
62 chars = ['a', 'b', 'c', 'd', 'x', 'y'];
|
wolffd@0
|
63 Osize = length(chars);
|
wolffd@0
|
64
|
wolffd@0
|
65 obsprob = zeros([Qsize Osize]);
|
wolffd@0
|
66 % 1 2 3 O
|
wolffd@0
|
67 obsprob(1,1,1,find(chars == 'a')) = 1.0;
|
wolffd@0
|
68
|
wolffd@0
|
69 obsprob(1,2,1,find(chars == 'x')) = 1.0;
|
wolffd@0
|
70 obsprob(1,2,2,find(chars == 'y')) = 1.0;
|
wolffd@0
|
71
|
wolffd@0
|
72 obsprob(1,3,1,find(chars == 'b')) = 1.0;
|
wolffd@0
|
73
|
wolffd@0
|
74 obsprob(2,1,1,find(chars == 'c')) = 1.0;
|
wolffd@0
|
75
|
wolffd@0
|
76 obsprob(2,2,1,find(chars == 'x')) = 1.0;
|
wolffd@0
|
77 obsprob(2,2,2,find(chars == 'y')) = 1.0;
|
wolffd@0
|
78
|
wolffd@0
|
79 obsprob(2,3,1,find(chars == 'd')) = 1.0;
|
wolffd@0
|
80
|
wolffd@0
|
81 Oargs = {'CPT', obsprob};
|
wolffd@0
|
82
|
wolffd@0
|
83 bnet = mk_hhmm('Qsizes', Qsize, 'Osize', Osize, 'discrete_obs', 1, ...
|
wolffd@0
|
84 'Oargs', Oargs, 'Ops', Qnodes(1:3), ...
|
wolffd@0
|
85 'startprob', startprob, 'transprob', transprob, 'termprob', termprob);
|
wolffd@0
|
86
|
wolffd@0
|
87
|
wolffd@0
|
88 Q1 = 1; Q2 = 2; Q3 = 3; F3 = 4; F2 = 5; Onode = 6;
|
wolffd@0
|
89 Qnodes = [Q1 Q2 Q3]; Fnodes = [F2 F3];
|
wolffd@0
|
90
|
wolffd@0
|
91 for seqi=1:3
|
wolffd@0
|
92 evidence = sample_dbn(bnet, 'stop_test', 'is_F2_true_D3');
|
wolffd@0
|
93 ev = cell2num(evidence);
|
wolffd@0
|
94 chars(ev(end,:))
|
wolffd@0
|
95 %T = size(evidence, 2)
|
wolffd@0
|
96 %pretty_print_hhmm_parse(evidence, Qnodes, Fnodes, Onode, chars);
|
wolffd@0
|
97 end
|