diff options
Diffstat (limited to 'libs/qm-dsp/hmm/hmm.h')
-rw-r--r-- | libs/qm-dsp/hmm/hmm.h | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/libs/qm-dsp/hmm/hmm.h b/libs/qm-dsp/hmm/hmm.h new file mode 100644 index 0000000000..da5c785bfa --- /dev/null +++ b/libs/qm-dsp/hmm/hmm.h @@ -0,0 +1,52 @@ +#ifndef _HMM_H +#define _HMM_H + +#ifdef __cplusplus +extern "C" { +#endif + +/* + * hmm.h + * + * Created by Mark Levy on 12/02/2006. + * Copyright 2006 Centre for Digital Music, Queen Mary, University of London. + + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License as + published by the Free Software Foundation; either version 2 of the + License, or (at your option) any later version. See the file + COPYING included with this distribution for more information. + * + */ + +#ifndef PI +#define PI 3.14159265358979323846264338327950288 +#endif + +typedef struct _model_t { + int N; /* number of states */ + double* p0; /* initial probs */ + double** a; /* transition probs */ + int L; /* dimensionality of data */ + double** mu; /* state means */ + double** cov; /* covariance, tied between all states */ +} model_t; + +void hmm_train(double** x, int T, model_t* model); /* with scaling */ +void forward_backwards(double*** xi, double** gamma, double* loglik, double* loglik1, double* loglik2, int iter, + int N, int T, double* p0, double** a, double** b); +void baum_welch(double* p0, double** a, double** mu, double** cov, int N, int T, int L, double** x, double*** xi, double** gamma); +void viterbi_decode(double** x, int T, model_t* model, int* q); /* using logs */ +model_t* hmm_init(double** x, int T, int L, int N); +void hmm_close(model_t* model); +void invert(double** cov, int L, double** icov, double* detcov); /* uses LAPACK (included with Mac OSX) */ +double gauss(double* x, int L, double* mu, double** icov, double detcov, double* y, double* z); +double loggauss(double* x, int L, double* mu, double** icov, double detcov, double* y, double* z); +void hmm_print(model_t* model); + +#ifdef __cplusplus +} +#endif + +#endif + |