comparison toolboxes/FullBNT-1.0.7/GraphViz/draw_hmm.m @ 0:e9a9cd732c1e tip

first hg version after svn

+function draw_hmm(A, varargin)
+% DRAW_HMM Make a picture of the HMM using dotty
+% function draw_hmm(A, ...)
+%
+% For details on dotty, see http://www.research.att.com/sw/tools/graphviz
+%
+% If A(i,j) > thresh, we draw and arc from state i to state j.
+%
+% Optional arguments (name/value pairs) [default]
+%
+% thresh - [1e-1]
+% obsprob - If B(i,o) > 0, we include "o" in the name of state i.
+%     e.g., if state 5 emits 1,3,7, its label becomes "5: 1 3 7".
+% startprob - ifstartprob(i) > 0, the state name will be prefixed with "+".
+% endprob - if endprob(i) > 0, the state name will be appended with "-".
+% filename - if [], we write to 'tmp.dot', convert this to 'tmp.ps'
+%  using 'dot -Tps tmp.dot -o tmp.ps', and then call ghostview  to display the result.
+%  dot and gv must be on your system path.
+%  If filename ~= [], we just generate the dot file, and do not
+%  convert it to postscript or call ghostview.
+
+[thresh, B, startprob, endprob, filename] = ...
+    process_options(varargin, 'thresh', 1e-1, 'obsprob', [], 'startprob', [], 'endprob', [], ...
+		    'filename', []);
+
+Q = length(A);
+
+arclabel = cell(Q,Q);
+G = zeros(Q,Q);
+for i=1:Q
+  for j=1:Q
+    if A(i,j) < thresh
+      arclabel{i,j} = '';
+    else
+      G(i,j) = 1;
+      arclabel{i,j} = sprintf('%5.3f', A(i,j));
+    end
+  end
+end
+
+
+nodelabel = cell(1,Q);
+for i=1:Q
+  % annotate start/stop states
+  if ~isempty(startprob) & ~approxeq(startprob(i), 0)
+    start = '+';
+  else
+    start = '';
+  end
+  if ~isempty(endprob) & ~approxeq(hmm.endprob(i), 0)
+    stop = '-';
+  else
+    stop = '';
+  end
+  label = sprintf('%s%d%s :', start, i, stop);
+
+  if ~isempty(B)
+    output_label = mk_output_label(B);
+    label = strcat(label, output_label);
+  end
+  
+  nodelabel{i} = label;
+end
+
+
+if isempty(filename)
+  filename = 'tmp.dot';
+  %mkdot(G, filename, arclabel, nodelabel)  
+  graph_to_dot(G, 'filename', filename, 'arc_label', arclabel, 'node_label', nodelabel);
+  fprintf('converting from .ps to .dot\n')
+  !dot -Tps tmp.dot -o tmp.ps
+  !gv tmp.ps &
+else
+  graph_to_dot(G, 'filename', filename, 'arc_label', arclabel, 'node_label', nodelabel);
+  %mkdot(G, filename, arclabel, nodelabel)  
+end
+
+
+%%%%%%%%%
+
+function label = mk_output_label(B)
+
+[Q O] = size(B);
+label = '';
+
+if 0
+  % print most probable symbols
+  for i=1:Q
+    m = max(B(i,:));
+    ndx = find(abs(B(i,:) - repmat(m,1,O)) < 1e-2);
+    %ndx = find(B(i,:)==m);
+    %label = sprintf('%d,', ndx);
+  end
+end
+
+if 0
+  % print prob distrib over all symbols 
+  for o=1:O
+    if approxeq(B(i,o), 0)
+      %
+    else
+      label = strcat(label, sprintf('%d(%3.2f),', o, B(i,o)));
+    end
+  end
+end
+
+if 1
+  % print all non-zero symbols
+  chars = ['a' 'b' 'c'];
+  for o=1:O
+    if approxeq(B(i,o), 0)
+      %
+    else
+      label = strcat(label, sprintf('%s', chars(o)));
+    end
+  end
+end