Daniel@0: seed = 1;
Daniel@0: rand('state', seed);
Daniel@0: randn('state', seed);
Daniel@0: 
Daniel@0: obs_model = 'unique';  % each cell has a unique label (essentially fully observable)
Daniel@0: %obs_model = 'four'; % each cell generates 4 observations, NESW
Daniel@0: 
Daniel@0: % Generate the true network, and a randomization of it
Daniel@0: realnet = mk_map_hhmm('p', 0.9, 'obs_model', obs_model);
Daniel@0: rndnet = mk_rnd_map_hhmm('obs_model', obs_model);
Daniel@0: eclass = realnet.equiv_class;
Daniel@0: U = 1; A = 2; C = 3; F = 4; onodes = 5;
Daniel@0: 
Daniel@0: ss = realnet.nnodes_per_slice;
Daniel@0: T = 100;
Daniel@0: evidence = sample_dbn(realnet, 'length', T);
Daniel@0: ev = cell(ss,T);
Daniel@0: ev(onodes,:) = evidence(onodes,:);
Daniel@0: 
Daniel@0: infeng = jtree_dbn_inf_engine(rndnet);
Daniel@0: 
Daniel@0: if 0
Daniel@0: % suppose we do not observe the final finish node, but only know 
Daniel@0: % it is more likely to be on that off
Daniel@0: ev2 = ev;
Daniel@0: infeng = enter_evidence(infeng, ev2, 'soft_evidence_nodes', [F T], 'soft_evidence',  {[0.3 0.7]'});
Daniel@0: end
Daniel@0: 
Daniel@0: 
Daniel@0: learnednet = learn_params_dbn_em(infeng, {evidence}, 'max_iter', 5);
Daniel@0: 
Daniel@0: disp('real model')
Daniel@0: disp_map_hhmm(realnet)
Daniel@0: 
Daniel@0: disp('learned model')
Daniel@0: disp_map_hhmm(learnednet)
Daniel@0: 
Daniel@0: disp('rnd model')
Daniel@0: disp_map_hhmm(rndnet)
Daniel@0: