bnet_to_fgraph.m

来自「麻省理工学院的人工智能工具箱,很珍贵,希望对大家有用!」· M 代码 · 共 17 行

function fg = bnet_to_fgraph(bnet)
% BNET_TO_FGRAPH Convert a Bayes net to a factor graph
% fg = bnet_to_fgraph(bnet)
%
% We create one factor per family, whose kernel is the CPD

nnodes = length(bnet.dag);
G = zeros(nnodes, nnodes);
for i=1:nnodes
  G(family(bnet.dag, i), i) = 1;
end

fg = mk_fgraph(G, bnet.node_sizes, bnet.CPD, 'equiv_class', bnet.equiv_class, 'discrete', bnet.dnodes);