jtree_onepass_inf_engine.m

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

function engine = jtree_onepass_inf_engine(bnet, query, obs_nodes, clusters)
% JTREE_ONEPASS_INF_ENGINE Junction tree inference engine for computing a single marginal
%
% engine = jtree_inf_engine(bnet, query)
% means we can compute the marginal on the nodes in 'query' in one-pass, using enter_evidence.
% This is just like variable-elimination, but can be more efficient, since we attempt to find a
% good elimination ordering. This is used by fast_jtree_dbn to compute alpha/beta efficiently.
%
% engine = jtree_onepass_inf_engine(bnet, query, obs_nodes, clusters)
% See jtree_inf_engine for more details of these arguments.

if nargin < 3, obs_nodes = []; end
if nargin < 4, clusters = {}; end

stages = { 1:length(bnet.dag) };
clusters{end+1} = query;

% parent class
pengine = jtree_inf_engine(bnet, obs_nodes, clusters, stages, query);

engine.query = query;
engine.marginal = [];

engine = class(engine, 'jtree_onepass_inf_engine', pengine);