diff options
Diffstat (limited to 'sourcecodes/bnt-master/BNT/learning')
| -rw-r--r-- | sourcecodes/bnt-master/BNT/learning/learn_params.m | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/sourcecodes/bnt-master/BNT/learning/learn_params.m b/sourcecodes/bnt-master/BNT/learning/learn_params.m index 1bf9c843..3d7ba847 100644 --- a/sourcecodes/bnt-master/BNT/learning/learn_params.m +++ b/sourcecodes/bnt-master/BNT/learning/learn_params.m @@ -21,6 +21,9 @@ for j=1:n fam = family(bnet.dag,j); %bnet.CPD{j} = learn_params(bnet.CPD{j}, data(fam,:)); bnet.CPD{j} = learn_params(bnet.CPD{j}, fam, data, bnet.node_sizes, bnet.cnodes); + if ismember(e,bnet.dnodes) + bnet.CPD{j} = learn_params_orig(bnet.CPD{j}, j, data, bnet.node_sizes, bnet.cnodes); + end end end |