annotate hmm/hmm.h @ 57:d241e7701c0c

* remove some debug output
author cannam
date Fri, 27 Feb 2009 13:07:22 +0000
parents 8bdbda7fb893
children 67899fda84f5
rev   line source
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