Mercurial > hg > camir-aes2014
comparison toolboxes/FullBNT-1.0.7/bnt/examples/dynamic/HHMM/Old/mk_abcd_hhmm.m @ 0:e9a9cd732c1e tip
first hg version after svn
author | wolffd |
---|---|
date | Tue, 10 Feb 2015 15:05:51 +0000 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:e9a9cd732c1e |
---|---|
1 % Make the HHMM in Figure 1 of the NIPS'01 paper | |
2 | |
3 Qsize = [2 3 2]; | |
4 D = 3; | |
5 | |
6 % transprob{d}(i,k,j), transprob{1}(i,j) | |
7 % termprob{d}(k,j), termprob{1}(1,j) | |
8 % startprob{d}(k,j), startprob{1}(1,j) | |
9 % obsprob(k, o) for discrete outputs | |
10 | |
11 % LEVEL 1 | |
12 % 1 2 e | |
13 A{1} = [0 0 1; | |
14 0 0 1]; | |
15 [transprob{1}, termprob{1}] = remove_hhmm_end_state(A{1}); | |
16 startprob{1} = [0.5 0.5]; | |
17 Q1args = {'startprob', startprob{1}, 'transprob', transprob{1}}; | |
18 | |
19 % LEVEL 2 | |
20 A{2} = zeros(Qsize(2), Qsize(1), Qsize(2)+1); | |
21 | |
22 % 1 2 3 e | |
23 A{2}(:,1,:) = [0 1 0 0 | |
24 0 0 1 0 | |
25 0 0 0 1]; | |
26 | |
27 % 1 2 3 e | |
28 A{2}(:,2,:) = [0 1 0 0 | |
29 0 0 1 0 | |
30 0 0 0 1]; | |
31 | |
32 [transprob{2}, termprob{2}] = remove_hhmm_end_state(A{2}); | |
33 | |
34 % always enter level 2 in state 1 | |
35 startprob{2} = [1 0 0 | |
36 1 0 0]; | |
37 | |
38 Q2args = {'startprob', startprob{2}, 'transprob', transprob{2}}; | |
39 F2args = {'CPT', termprob{2}}; | |
40 | |
41 | |
42 % LEVEL 3 | |
43 | |
44 A{3} = zeros([Qsize(3) Qsize(1:2) Qsize(3)+1]); | |
45 endstate = Qsize(3)+1; | |
46 % Qt-1(3) Qt(1) Qt(2) Qt(3) | |
47 % 1 2 e | |
48 A{3}(1, 1, 1, endstate) = 1.0; | |
49 A{3}(:, 1, 2, :) = [0.0 1.0 0.0 | |
50 0.5 0.0 0.5]; | |
51 A{3}(1, 1, 3, endstate) = 1.0; | |
52 | |
53 A{3}(1, 2, 1, endstate) = 1.0; | |
54 A{3}(:, 2, 2, :) = [0.0 1.0 0.0 | |
55 0.5 0.0 0.5]; | |
56 A{3}(1, 2, 3, endstate) = 1.0; | |
57 | |
58 A{3} = reshape(A{3}, [Qsize(3) prod(Qsize(1:2)) Qsize(3)+1]); | |
59 [transprob{3}, termprob{3}] = remove_hhmm_end_state(A{3}); | |
60 | |
61 % define the vertical entry points to level 3 | |
62 startprob{3} = zeros(Qsize); | |
63 % Q1 Q2 Q3 | |
64 startprob{3}(1, 1, 1) = 1.0; | |
65 startprob{3}(1, 2, 1) = 1.0; | |
66 startprob{3}(1, 3, 1) = 1.0; | |
67 | |
68 startprob{3}(2, 1, 1) = 1.0; | |
69 startprob{3}(2, 2, 1) = 1.0; | |
70 startprob{3}(2, 3, 1) = 1.0; | |
71 | |
72 startprob{3} = reshape(startprob{3}, prod(Qsize(1:2)), Qsize(3)); | |
73 | |
74 chars = ['a', 'b', 'c', 'd', 'x', 'y']; | |
75 Osize = length(chars); | |
76 | |
77 obsprob = zeros([Qsize Osize]); | |
78 % 1 2 3 O | |
79 obsprob(1,1,1,find(chars == 'a')) = 1.0; | |
80 | |
81 obsprob(1,2,1,find(chars == 'x')) = 1.0; | |
82 obsprob(1,2,2,find(chars == 'y')) = 1.0; | |
83 | |
84 obsprob(1,3,1,find(chars == 'b')) = 1.0; | |
85 | |
86 obsprob(2,1,1,find(chars == 'c')) = 1.0; | |
87 | |
88 obsprob(2,2,1,find(chars == 'x')) = 1.0; | |
89 obsprob(2,2,2,find(chars == 'y')) = 1.0; | |
90 | |
91 obsprob(2,3,1,find(chars == 'd')) = 1.0; | |
92 | |
93 obsprob = reshape(obsprob, prod(Qsize), Osize); | |
94 | |
95 [intra, inter, Qnodes, Fnodes, Onode] = mk_hhmm_topo(D); | |
96 | |
97 hhmm.Qnodes = Qnodes; | |
98 hhmm.Fnodes = Fnodes; | |
99 hhmm.Onode = Onode; | |
100 hhmm.D = D; | |
101 hhmm.Qsize = Qsize; | |
102 hhmm.Osize = Osize; | |
103 hhmm.startprob = startprob; | |
104 hhmm.transprob = transprob; | |
105 hhmm.termprob = termprob; | |
106 hhmm.obsprob = obsprob; | |
107 hhmm.A = A; | |
108 | |
109 |