Daniel@0: function bnet = mk_asia_bnet(CPD_type, p, arity)
Daniel@0: % MK_ASIA_BNET Make the 'Asia' bayes net.
Daniel@0: %
Daniel@0: % BNET = MK_ASIA_BNET uses the parameters specified on p21 of Cowell et al, 
Daniel@0: % "Probabilistic networks and expert systems", Springer Verlag 1999.
Daniel@0: % 
Daniel@0: % BNET = MK_ASIA_BNET('cpt', p) uses random parameters drawn from a Dirichlet(p,p,...)
Daniel@0: % distribution. If p << 1, this is nearly deterministic; if p >> 1, this is nearly uniform.
Daniel@0: % 
Daniel@0: % BNET = MK_ASIA_BNET('bool') makes each CPT a random boolean function.
Daniel@0: %
Daniel@0: % BNET = MK_ASIA_BNET('gauss') makes each CPT a random linear Gaussian distribution.
Daniel@0: %
Daniel@0: % BNET = MK_ASIA_BNET('orig') is the same as MK_ASIA_BNET.
Daniel@0: %
Daniel@0: % BNET = MK_ASIA_BNET('cpt', p, arity) can specify non-binary nodes.
Daniel@0: 
Daniel@0: 
Daniel@0: if nargin == 0, CPD_type = 'orig'; end
Daniel@0: if nargin < 3, arity = 2; end
Daniel@0: 
Daniel@0: Smoking = 1;
Daniel@0: Bronchitis = 2;
Daniel@0: LungCancer = 3;
Daniel@0: VisitToAsia = 4;
Daniel@0: TB = 5;
Daniel@0: TBorCancer = 6;
Daniel@0: Dys = 7;
Daniel@0: Xray = 8;
Daniel@0: 
Daniel@0: n = 8;
Daniel@0: dag = zeros(n);
Daniel@0: dag(Smoking, [Bronchitis LungCancer]) = 1;
Daniel@0: dag(Bronchitis, Dys) = 1;
Daniel@0: dag(LungCancer, TBorCancer) = 1;
Daniel@0: dag(VisitToAsia, TB) = 1;
Daniel@0: dag(TB, TBorCancer) = 1;
Daniel@0: dag(TBorCancer, [Dys Xray]) = 1;
Daniel@0: 
Daniel@0: ns = arity*ones(1,n);
Daniel@0: if strcmp(CPD_type, 'gauss')
Daniel@0:   dnodes = [];
Daniel@0: else
Daniel@0:   dnodes = 1:n;
Daniel@0: end
Daniel@0: bnet = mk_bnet(dag, ns, 'discrete', dnodes);
Daniel@0: 
Daniel@0: switch CPD_type
Daniel@0:   case 'orig', 
Daniel@0:     % true is 2, false is 1
Daniel@0:     bnet.CPD{VisitToAsia} = tabular_CPD(bnet, VisitToAsia, [0.99   0.01]);
Daniel@0:     bnet.CPD{Bronchitis} = tabular_CPD(bnet, Bronchitis, [0.7 0.4   0.3 0.6]);
Daniel@0:     % minka: bug fix
Daniel@0:     bnet.CPD{Dys} = tabular_CPD(bnet, Dys, [0.9 0.2 0.3 0.1   0.1 0.8 0.7 0.9]);
Daniel@0:     bnet.CPD{TBorCancer} = tabular_CPD(bnet, TBorCancer, [1 0 0 0   0 1 1 1]);
Daniel@0:     % minka: bug fix
Daniel@0:     bnet.CPD{LungCancer} = tabular_CPD(bnet, LungCancer, [0.99 0.9  0.01 0.1]);
Daniel@0:     bnet.CPD{Smoking} = tabular_CPD(bnet, Smoking, [0.5 0.5]);
Daniel@0:     bnet.CPD{TB} = tabular_CPD(bnet, TB, [0.99 0.95  0.01 0.05]);
Daniel@0:     bnet.CPD{Xray} = tabular_CPD(bnet, Xray, [0.95 0.02  0.05 0.98]);
Daniel@0:  case 'bool',
Daniel@0:   for i=1:n
Daniel@0:     bnet.CPD{i} = boolean_CPD(bnet, i, 'rnd');
Daniel@0:   end
Daniel@0:  case 'gauss',
Daniel@0:   for i=1:n
Daniel@0:     bnet.CPD{i} = gaussian_CPD(bnet, i, 'cov', 1*eye(ns(i)));
Daniel@0:   end
Daniel@0:  case 'cpt',
Daniel@0:   for i=1:n
Daniel@0:     bnet.CPD{i} = tabular_CPD(bnet, i, p);
Daniel@0:   end
Daniel@0: end
Daniel@0: 
Daniel@0:   
Daniel@0: