Mercurial > hg > camir-aes2014
comparison toolboxes/FullBNT-1.0.7/bnt/inference/online/@jtree_sparse_2TBN_inf_engine/fwd.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 function [f, logscale] = fwd(engine, fpast, ev, t) | |
2 % Forwards pass. | |
3 | |
4 bnet = bnet_from_engine(engine); | |
5 ss = bnet.nnodes_per_slice; | |
6 | |
7 ev2 = cell(ss, 2); | |
8 ev2(:,1) = fpast.evidence; | |
9 ev2(:,2) = ev; | |
10 CPDpot = cell(1,ss); | |
11 for n=1:ss | |
12 fam = family(bnet.dag, n, 2); | |
13 e = bnet.equiv_class(n, 2); | |
14 CPDpot{n} = convert_to_pot(bnet.CPD{e}, engine.pot_type, fam(:), ev2); | |
15 end | |
16 f.evidence = ev; | |
17 f.t = t; | |
18 | |
19 % get prior | |
20 int = engine.interface; | |
21 if fpast.t==1 | |
22 prior = marginalize_pot(fpast.clpot{engine.int_clq1}, int, engine.maximize); | |
23 else | |
24 prior = marginalize_pot(fpast.clpot{engine.out_clq}, int+ss, engine.maximize); | |
25 prior = set_domain_pot(prior, int); % shift back to slice 1 | |
26 end | |
27 | |
28 pots = [ {prior} CPDpot ]; | |
29 slice1 = 1:ss; | |
30 slice2 = slice1 + ss; | |
31 CPDclqs = engine.clq_ass_to_node(slice2); | |
32 D = engine.in_clq; | |
33 clqs = [D CPDclqs]; | |
34 | |
35 [f.clpot, f.seppot] = init_pot(engine.jtree_engine, clqs, pots, engine.pot_type, engine.observed); | |
36 [f.clpot, f.seppot] = collect_evidence(engine.jtree_engine, f.clpot, f.seppot); | |
37 for c=1:length(f.clpot) | |
38 if isa(f.clpot{c}, 'struct') | |
39 domain = f.clpot{c}.domain; | |
40 sizes = f.clpot{c}.sizes; | |
41 T = f.clpot{c}.T; | |
42 f.clpot{c} = dpot(domain, sizes, T); | |
43 end | |
44 [f.clpot{c}, ll(c)] = normalize_pot(f.clpot{c}); | |
45 end | |
46 logscale = ll(engine.root); | |
47 |