cannam@16
|
1 #ifndef _HMM_H
|
cannam@16
|
2 #define _HMM_H
|
cannam@16
|
3
|
cannam@20
|
4 #ifdef __cplusplus
|
cannam@20
|
5 extern "C" {
|
cannam@20
|
6 #endif
|
cannam@20
|
7
|
cannam@16
|
8 /*
|
cannam@16
|
9 * hmm.h
|
cannam@16
|
10 * soundbite
|
cannam@16
|
11 *
|
cannam@16
|
12 * Created by Mark Levy on 12/02/2006.
|
cannam@16
|
13 * Copyright 2006 Centre for Digital Music, Queen Mary, University of London. All rights reserved.
|
cannam@16
|
14 *
|
cannam@16
|
15 */
|
cannam@16
|
16
|
cannam@16
|
17 #ifndef PI
|
cannam@16
|
18 #define PI 3.14159265358979323846264338327950288
|
cannam@16
|
19 #endif
|
cannam@16
|
20
|
cannam@16
|
21 typedef struct _model_t {
|
cannam@16
|
22 int N; /* number of states */
|
cannam@16
|
23 double* p0; /* initial probs */
|
cannam@16
|
24 double** a; /* transition probs */
|
cannam@16
|
25 int L; /* dimensionality of data */
|
cannam@16
|
26 double** mu; /* state means */
|
cannam@16
|
27 double** cov; /* covariance, tied between all states */
|
cannam@16
|
28 } model_t;
|
cannam@16
|
29
|
cannam@16
|
30 void hmm_train(double** x, int T, model_t* model); /* with scaling */
|
cannam@16
|
31 void forward_backwards(double*** xi, double** gamma, double* loglik, double* loglik1, double* loglik2, int iter,
|
cannam@16
|
32 int N, int T, double* p0, double** a, double** b);
|
cannam@16
|
33 void baum_welch(double* p0, double** a, double** mu, double** cov, int N, int T, int L, double** x, double*** xi, double** gamma);
|
cannam@16
|
34 void viterbi_decode(double** x, int T, model_t* model, int* q); /* using logs */
|
cannam@16
|
35 model_t* hmm_init(double** x, int T, int L, int N);
|
cannam@16
|
36 void hmm_close(model_t* model);
|
cannam@16
|
37 void invert(double** cov, int L, double** icov, double* detcov); /* uses LAPACK (included with Mac OSX) */
|
cannam@16
|
38 double gauss(double* x, int L, double* mu, double** icov, double detcov, double* y, double* z);
|
cannam@16
|
39 double loggauss(double* x, int L, double* mu, double** icov, double detcov, double* y, double* z);
|
cannam@16
|
40 void hmm_print(model_t* model);
|
cannam@16
|
41
|
cannam@20
|
42 #ifdef __cplusplus
|
cannam@20
|
43 }
|
cannam@16
|
44 #endif
|
cannam@16
|
45
|
cannam@20
|
46 #endif
|
cannam@20
|
47
|