diff hmm/hmm.h @ 241:a98dd8ec96f8

* Move dsp/maths to maths ; bring PCA and HMM across from Soundbite
author Chris Cannam <c.cannam@qmul.ac.uk>
date Wed, 09 Jan 2008 10:31:29 +0000
parents
children 8bdbda7fb893
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hmm/hmm.h	Wed Jan 09 10:31:29 2008 +0000
@@ -0,0 +1,39 @@
+#ifndef _HMM_H
+#define _HMM_H
+
+/*
+ *  hmm.h
+ *  soundbite
+ *
+ *  Created by Mark Levy on 12/02/2006.
+ *  Copyright 2006 Centre for Digital Music, Queen Mary, University of London. All rights reserved.
+ *
+ */
+
+#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);
+
+#endif
+