marginal_nodes.m

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

function marginal = marginal_nodes(engine, query)
% MARGINAL_NODES Compute the marginal on the specified query nodes (cond_gauss)
% marginal = marginal_nodes(engine, query)
%
% 'query' must be a singleton set.

assert(length(query)==1);
j = query;
bnet = bnet_from_engine(engine);

if myismember(j, bnet.cnodes)
  if ~myismember(j, engine.onodes)
    [m, C] = collapse_mog(engine.mu{j}, engine.Sigma{j}, engine.T);    
    marginal.mu = m;
    marginal.Sigma = C;
    marginal.T = 1.0; % since P(E|E)=1
  else
    marginal.mu = [];
    marginal.Sigma = [];
    marginal.T = 1.0;
  end
else
  marginal = pot_to_marginal(marginalize_pot(engine.joint_dmarginal, j));
end