blob: cd17a6234048103fc566ec10c73891403380fc99 (
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
29
30
31
32
|
function [mpe, prob] = calc_mpe_given_inf_engine(engine, evidence)
% CALC_MPE_GIVEN_ENGINE Computes the most probable explanation of the evidence
% [mpe, prob] = calc_mpe_given_inf_engine(engine, evidence)
%
% INPUT
% engine must support max-propagation
% evidence{i} is the obsevred value of node i, or [] if hidden
%
% OUTPUT
% mpe(i) is the most likely value of node i
% prob is the likelihood of the globally best assignment
%
% This currently only works when all nodes are discrete
[engine, ll] = enter_evidence(engine, evidence);
observed = ~isemptycell(evidence);
N = length(evidence);
mpe = zeros(1,N);
for i=1:N
m = marginal_nodes(engine, i);
% discrete observed nodes are all set to 1 inside the inference engine, so we must undo this
if observed(i)
mpe(i) = evidence{i};
else
mpe(i) = argmax(m.T);
end
end
bnet = bnet_from_engine(engine);
ll = log_lik_complete(bnet, num2cell(mpe(:)));
prob = exp(ll);
|
