Daniel@0: % Compare the speeds of various inference engines on a coupled HMM
Daniel@0: 
Daniel@0: N = 3;
Daniel@0: Q = 2;
Daniel@0: rand('state', 0);
Daniel@0: randn('state', 0);
Daniel@0: discrete = 0;
Daniel@0: if discrete
Daniel@0:   Y = 2; % size of output alphabet
Daniel@0: else
Daniel@0:   Y = 3; % size of observed vectors
Daniel@0: end
Daniel@0: coupled = 1;
Daniel@0: bnet = mk_chmm(N, Q, Y, discrete, coupled); 
Daniel@0: %bnet = mk_fhmm(N, Q, Y, discrete);  % factorial HMM
Daniel@0: ss = length(bnet.node_sizes_slice);
Daniel@0: 
Daniel@0: T = 3;
Daniel@0: 
Daniel@0: USEC = exist('@jtree_C_inf_engine/collect_evidence','file');
Daniel@0: 
Daniel@0: engine = {};
Daniel@0: engine{end+1} = jtree_dbn_inf_engine(bnet);
Daniel@0: %engine{end+1} = jtree_ndx_dbn_inf_engine(bnet, 'ndx_type', 'SD');
Daniel@0: %engine{end+1} = jtree_ndx_dbn_inf_engine(bnet, 'ndx_type', 'D');
Daniel@0: %engine{end+1} = jtree_ndx_dbn_inf_engine(bnet, 'ndx_type', 'B');
Daniel@0: if USEC, engine{end+1} = jtree_C_dbn_inf_engine(bnet); end
Daniel@0: engine{end+1} = hmm_inf_engine(bnet);
Daniel@0: engine{end+1} = jtree_unrolled_dbn_inf_engine(bnet, T); 
Daniel@0: 
Daniel@0: % times in matlab N=4 Q=4 T=5 (* = winner)
Daniel@0: %     jtree    SD       B          hmm       dhmm      unrolled
Daniel@0: %    0.6266    1.1563    8.3815    0.3069    0.1948*    0.8654  inf
Daniel@0: %    0.9057*   2.1522   12.6314    2.6847    2.3107    3.1905  learn
Daniel@0: 
Daniel@0: %engine{end+1} = bk_inf_engine(bnet, 'ff', onodes);
Daniel@0: %engine{end+1} = pearl_unrolled_dbn_inf_engine(bnet, T);
Daniel@0: 
Daniel@0: inf_time = cmp_inference_dbn(bnet, engine, T)
Daniel@0: learning_time = cmp_learning_dbn(bnet, engine, T)
Daniel@0: