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