Mercurial > hg > camir-aes2014
diff toolboxes/FullBNT-1.0.7/bnt/general/enumerate_scenarios.m @ 0:e9a9cd732c1e tip
first hg version after svn
author | wolffd |
---|---|
date | Tue, 10 Feb 2015 15:05:51 +0000 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/toolboxes/FullBNT-1.0.7/bnt/general/enumerate_scenarios.m Tue Feb 10 15:05:51 2015 +0000 @@ -0,0 +1,21 @@ +function [scenarios, log_probs] = enumerate_scenarios(bnet, evidence) +% ENUMERATE_SCENARIOS Enumerate all assignments, and return the prob. of the non-zeros ones +% function [scenarios, log_probs] = enumerate_scenarios(bnet, evidence) + +assert(isempty(bnet.cnodes)); +n = length(bnet.dag); +observed = ~isemptycell(evidence); +vals = cat(1,evidence{observed}); +vals = vals(:)'; +ns = bnet.node_sizes; + +log_probs = []; +scenarios = []; +for i=1:prod(ns) + inst = ind2subv(ns, i); % i'th instantiation + if isempty(vals) | inst(observed) == vals % agrees with evidence + ll = log_lik_complete(bnet, num2cell(inst(:))); + log_probs = [log_probs ll]; + scenarios = [scenarios(:)' inst]; + end +end