camir-aes2014: toolboxes/FullBNT-1.0.7/bnt/inference/static/@cond_gauss_inf_engine/marginal

first hg version after svn

rev	line source
wolffd@0	1 function marginal = marginal_nodes(engine, query, add_ev)
wolffd@0	2 % MARGINAL_NODES Compute the marginal on the specified query nodes (cond_gauss)
wolffd@0	3 % marginal = marginal_nodes(engine, query, add_ev)
wolffd@0	4 %
wolffd@0	5 % 'query' must be a singleton set
wolffd@0	6 % add_ev is an optional argument; if 1, we will "inflate" the marginal of observed nodes
wolffd@0	7 % to their original size, adding 0s to the positions which contradict the evidence
wolffd@0	8
wolffd@0	9 if nargin < 3, add_ev = 0; end
wolffd@0	10
wolffd@0	11 if length(query) ~= 1
wolffd@0	12 error('cond_gauss_inf_engine can only handle marginal queries on single nodes')
wolffd@0	13 end
wolffd@0	14 j = query;
wolffd@0	15 bnet = bnet_from_engine(engine);
wolffd@0	16
wolffd@0	17 if myismember(j, bnet.cnodes)
wolffd@0	18 if ~myismember(j, engine.onodes)
wolffd@0	19 [m, C] = collapse_mog(engine.mu{j}, engine.Sigma{j}, engine.T);
wolffd@0	20 marginal.mu = m;
wolffd@0	21 marginal.Sigma = C;
wolffd@0	22 marginal.T = 1.0; % single mixture component
wolffd@0	23 else
wolffd@0	24 marginal.mu = engine.evidence{j};
wolffd@0	25 k = bnet.node_sizes(j);
wolffd@0	26 marginal.Sigma = zeros(k,k);
wolffd@0	27 marginal.T = 1.0; % since P(E\|E)=1
wolffd@0	28 end
wolffd@0	29 else
wolffd@0	30 marginal = pot_to_marginal(marginalize_pot(engine.joint_dmarginal, j));
wolffd@0	31 if add_ev
wolffd@0	32 marginal = add_ev_to_dmarginal(marginal, engine.evidence, bnet.node_sizes);
wolffd@0	33 end
wolffd@0	34 end
wolffd@0	35
wolffd@0	36 marginal.domain = query;

Mercurial > hg > camir-aes2014