about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/BNT/inference/static/@global_joint_inf_engine/find_mpe.m
blob: 92915b6cfec92836aa59fe79e2e209ec12ce9f00 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
function [mpe, ll] = find_mpe(engine, evidence)
% FIND_MPE_GLOBAL Compute the most probable explanation(s) from the global joint
% [mpe, ll] = find_mpe(engine, evidence)
%
% mpe(k,i) is the most probable value of node i in the k'th global mode  (cell array)
%
% We assume all nodes are discrete

%engine = global_joint_inf_engine(bnet);
bnet = bnet_from_engine(engine);
engine = enter_evidence(engine, evidence);
S1 = struct(engine); % violate object privacy
S2 = struct(S1.jpot); % joint potential
prob = max(S2.T(:));
modes = find(S2.T(:) == prob);

ens = bnet.node_sizes;
onodes = find(~isemptycell(evidence));
ens(onodes) = 1;
mpe = ind2subv(ens, modes);
for k=1:length(modes)
  for i=onodes(:)'
    mpe(k,i) = evidence{i};
  end
end
ll = log(prob);

mpe = num2cell(mpe);