Mercurial > hg > camir-aes2014
comparison toolboxes/FullBNT-1.0.7/bnt/inference/static/@jtree_inf_engine/find_max_config.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 [mpe, clpot, seppot] = find_max_config(engine, clpot, seppot, evidence) | |
2 % FIND_MAX_CONFIG Backwards pass of Viterbi fro jtree | |
3 % function [mpe, clpot, seppot] = find_max_config(engine, clpot, seppot, evidence) | |
4 % See Cowell99 p98 | |
5 | |
6 bnet = bnet_from_engine(engine); | |
7 nnodes = length(bnet.dag); | |
8 mpe = cell(1, nnodes); | |
9 maximize = 1; | |
10 | |
11 c = engine.root_clq; | |
12 pot = struct(clpot{c}); % violate object privacy | |
13 dom = pot.domain; | |
14 [indices, clpot{c}] = find_most_prob_entry(clpot{c}); | |
15 mpe(dom) = num2cell(indices); | |
16 | |
17 for n=engine.preorder | |
18 for c=engine.preorder_children{n} | |
19 clpot{c} = divide_by_pot(clpot{c}, seppot{n,c}); | |
20 seppot{n,c} = marginalize_pot(clpot{n}, engine.separator{n,c}, maximize); | |
21 clpot{c} = multiply_by_pot(clpot{c}, seppot{n,c}); | |
22 | |
23 pot = struct(clpot{c}); % violate object privacy | |
24 dom = pot.domain; | |
25 [indices, clpot{c}] = find_most_prob_entry(clpot{c}); | |
26 mpe(dom) = num2cell(indices); | |
27 end | |
28 end | |
29 | |
30 obs_nodes = find(~isemptycell(evidence)); | |
31 % indices for observed nodes will be 1 - need to overwrite these | |
32 mpe(obs_nodes) = evidence(obs_nodes); | |
33 | |
34 | |
35 |