diff options
Diffstat (limited to 'sourcecodes/bnt-master/BNT/general/bnet_to_fgraph.m')
| -rw-r--r-- | sourcecodes/bnt-master/BNT/general/bnet_to_fgraph.m | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/sourcecodes/bnt-master/BNT/general/bnet_to_fgraph.m b/sourcecodes/bnt-master/BNT/general/bnet_to_fgraph.m new file mode 100644 index 00000000..fa2763b9 --- /dev/null +++ b/sourcecodes/bnt-master/BNT/general/bnet_to_fgraph.m @@ -0,0 +1,16 @@ +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); + + + |