annotate toolboxes/FullBNT-1.0.7/bnt/inference/static/@enumerative_inf_engine/marginal_nodes.m @ 0:cc4b1211e677 tip

initial commit to HG from Changeset: 646 (e263d8a21543) added further path and more save "camirversion.m"
author Daniel Wolff
date Fri, 19 Aug 2016 13:07:06 +0200
parents
children
rev   line source
Daniel@0 1 function [marginal, loglik] = marginal_nodes(engine, query)
Daniel@0 2 % MARGINAL_NODES Compute the marginal on the specified query nodes (enumerative_inf)
Daniel@0 3 % [marginal, loglik] = marginal_nodes(engine, query)
Daniel@0 4
Daniel@0 5
Daniel@0 6 if isempty(query) & nargout < 2
Daniel@0 7 marginal.T = 1;
Daniel@0 8 marginal.domain = [];
Daniel@0 9 return;
Daniel@0 10 end
Daniel@0 11
Daniel@0 12 evidence = engine.evidence;
Daniel@0 13 bnet = bnet_from_engine(engine);
Daniel@0 14 assert(isempty(bnet.cnodes));
Daniel@0 15 n = length(bnet.dag);
Daniel@0 16 observed = ~isemptycell(evidence);
Daniel@0 17 vals = cat(1,evidence{observed});
Daniel@0 18 vals = vals(:)';
Daniel@0 19 ns = bnet.node_sizes;
Daniel@0 20
Daniel@0 21 sz = ns(query);
Daniel@0 22 T = 0*myones(sz);
Daniel@0 23 p = 0;
Daniel@0 24 for i=1:prod(ns)
Daniel@0 25 inst = ind2subv(ns, i); % i'th instantiation
Daniel@0 26 if isempty(vals) | inst(observed) == vals % agrees with evidence
Daniel@0 27 prob = exp(log_lik_complete(bnet, num2cell(inst(:))));
Daniel@0 28 p = p + prob;
Daniel@0 29 v = inst(query);
Daniel@0 30 j = subv2ind(sz, v);
Daniel@0 31 T(j) = T(j) + prob;
Daniel@0 32 end
Daniel@0 33 end
Daniel@0 34
Daniel@0 35 [T, lik] = normalise(T);
Daniel@0 36 lik = p;
Daniel@0 37 loglik = log(lik);
Daniel@0 38
Daniel@0 39 Tsmall = shrink_obs_dims_in_table(T, query, evidence);
Daniel@0 40 marginal.domain = query;
Daniel@0 41 marginal.T = Tsmall;