Daniel@0: function bnet = mk_water_dbn(discrete_obs, obs_leaves)
Daniel@0: % MK_WATER_DBN
Daniel@0: % bnet = mk_water_dbn(discrete_obs, obs_leaves)
Daniel@0: %
Daniel@0: % If discrete_obs = 1 (default), the leaves are binary, else scalar Gaussians
Daniel@0: % If obs_leaves = 1, all the leaves are observed, otherwise rnd nodes are observed
Daniel@0: %
Daniel@0: % This is a model of the biological processes of a water purification plant, developed
Daniel@0: % by Finn V. Jensen, Uffe Kjærulff, Kristian G. Olesen, and Jan Pedersen.
Daniel@0: % See http://www-nt.cs.berkeley.edu/home/nir/public_html/Repository/water.htm
Daniel@0: % See also Boyen and Koller, "Tractable Inference for Complex Stochastic Processes", UAI98
Daniel@0: 
Daniel@0: if nargin < 1, discrete_obs = 1; end
Daniel@0: if nargin < 1, obs_leaves = 1; end
Daniel@0: 
Daniel@0: ss = 12;
Daniel@0: intra = zeros(ss);
Daniel@0: intra(1,9) = 1;
Daniel@0: intra(3,10) = 1;
Daniel@0: intra(4,11) = 1;
Daniel@0: intra(8,12) = 1;
Daniel@0: 
Daniel@0: inter = zeros(ss);
Daniel@0: inter(1, [1 3]) = 1;
Daniel@0: inter(2, [2 3 7]) = 1;
Daniel@0: inter(3, [3 4 5]) = 1;
Daniel@0: inter(4, [3 4 6]) = 1;
Daniel@0: inter(5, [3 5 6]) = 1;
Daniel@0: inter(6, [4 5 6]) = 1;
Daniel@0: inter(7, [7 8]) = 1;
Daniel@0: inter(8, [6 7 8]) = 1;
Daniel@0: 
Daniel@0: if obs_leaves
Daniel@0:   onodes = 9:12; % leaves
Daniel@0: else
Daniel@0:   onodes = [1 5 9:12]; % throw in some other nodes
Daniel@0: end
Daniel@0: hnodes = 1:8;
Daniel@0: if discrete_obs
Daniel@0:   ns = 2*ones(1 ,ss);
Daniel@0:   dnodes = 1:ss;
Daniel@0: else
Daniel@0:   ns = [2*ones(1,length(hnodes)) 1*ones(length(onodes))];
Daniel@0:   dnodes = hnodes;
Daniel@0: end
Daniel@0: 
Daniel@0: eclass1 = 1:12;
Daniel@0: eclass2 = [13:20 9:12];
Daniel@0: bnet = mk_dbn(intra, inter, ns, 'discrete', dnodes, 'eclass1', eclass1, 'eclass2', eclass2, ...
Daniel@0: 	      'observed', onodes);
Daniel@0: if discrete_obs
Daniel@0:   for i=1:max(eclass2)
Daniel@0:     bnet.CPD{i} = tabular_CPD(bnet, i);
Daniel@0:   end
Daniel@0: else
Daniel@0:   for i=hnodes(:)'
Daniel@0:     bnet.CPD{i} = tabular_CPD(bnet, i);
Daniel@0:   end
Daniel@0:   for i=onodes(:)'
Daniel@0:     bnet.CPD{i} = gaussian_CPD(bnet, i);
Daniel@0:   end
Daniel@0:   for i=hnodes(:)'+ss
Daniel@0:     bnet.CPD{i} = tabular_CPD(bnet, i);
Daniel@0:   end
Daniel@0: end
Daniel@0: 
Daniel@0: 
Daniel@0: