1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
function [initState, transmat, mu, Sigma] = gausshmm_train_observed(obsData, hiddenData, ...
nstates, varargin)
% GAUSSHMM_TRAIN_OBSERVED Estimate params of HMM with Gaussian output from fully observed sequences
% [initState, transmat, mu, Sigma] = gausshmm_train_observed(obsData, hiddenData, nstates,...)
%
% INPUT
% If all sequences have the same length
% obsData(:,t,ex)
% hiddenData(ex,t) - must be ROW vector if only one sequence
% If sequences have different lengths, we use cell arrays
% obsData{ex}(:,t)
% hiddenData{ex}(t)
%
% Optional argumnets
% dirichletPriorWeight - for smoothing transition matrix counts
%
% Optional parameters from mixgauss_Mstep:
% 'cov_type' - 'full', 'diag' or 'spherical' ['full']
% 'tied_cov' - 1 (Sigma) or 0 (Sigma_i) [0]
% 'clamped_cov' - pass in clamped value, or [] if unclamped [ [] ]
% 'clamped_mean' - pass in clamped value, or [] if unclamped [ [] ]
% 'cov_prior' - Lambda_i, added to YY(:,:,i) [0.01*eye(d,d,Q)]
%
% Output
% mu(:,q)
% Sigma(:,:,q)
[dirichletPriorWeight, other] = process_options(...
varargin, 'dirichletPriorWeight', 0);
[transmat, initState] = transmat_train_observed(hiddenData, nstates, ...
'dirichletPriorWeight', dirichletPriorWeight);
% convert to obsData(:,t*nex)
if ~iscell(obsData)
[D T Nex] = size(obsData);
obsData = reshape(obsData, D, T*Nex);
else
obsData = cat(2, obsData{:});
hiddenData = cat(2,hiddenData{:});
end
[mu, Sigma] = condgaussTrainObserved(obsData, hiddenData(:), nstates, varargin{:});
|
