enter_evidence.m

来自「Bayes网络工具箱」· M 代码 · 共 38 行

function engine = enter_evidence(engine, evidence, nsamples)
% ENTER_EVIDENCE Add the specified evidence to the network (likelihood_weighting)
% engine = enter_evidence(engine, evidence, nsamples)
% evidence{i} = [] if if X(i) is hidden, and otherwise contains its observed value (scalar or column vector)
%
% If nsamples is not specified, the value specified when the engine was created will be used.

if nargin < 3, nsamples = engine.nsamples; end

bnet = bnet_from_engine(engine);
N = length(bnet.dag);
samples = cell(nsamples, N);
weights = zeros(1, nsamples);

ns = bnet.node_sizes;
original_evidence = evidence;
observed = ~isemptycell(original_evidence);
for s=1:nsamples
  evidence = original_evidence;
  w = 1;
  for i=1:N
    ps = parents(bnet.dag, i);
    e = bnet.equiv_class(i);
    if observed(i)
      p = evaluate_CPD(bnet.CPD{e}, [ps i], ns, bnet.cnodes, evidence);
      w = w * p;
    else
      x = sample_CPD(bnet.CPD{e}, [ps i], ns, bnet.cnodes, evidence);
      evidence{i} = x;
    end
  end
  samples(s,:) = evidence;
  weights(s) = w;
end                 

engine.samples = samples;
engine.weights = weights;