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