about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/BNT/CPDs/@gmux_CPD/CPD_to_pi.m
blob: 63b5726bd1c44aa14adcf8bb47d15c69acc98fc6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
function pi = CPD_to_pi(CPD, msg_type, n, ps, msg, evidence)
% CPD_TO_PI Compute the pi vector (gaussian)
% function pi = CPD_to_pi(CPD, msg_type, n, ps, msg, evidence)

switch msg_type
 case 'd',
  error('gaussian_CPD can''t create discrete msgs')
 case 'g',
  dps = ps(CPD.dps);
  k = evidence{dps};
  if isempty(k)
    error('gmux node must have observed discrete parent')
  end
  m = msg{n}.pi_from_parent{k}; 
  B = CPD.weights(:,:,k);
  pi.mu = CPD.mean(:,k) + B * m.mu;
  pi.Sigma = CPD.cov(:,:,k) + B * m.Sigma * B';
end