summaryrefslogtreecommitdiff
path: root/libs/qm-dsp/hmm/hmm.h
diff options
context:
space:
mode:
Diffstat (limited to 'libs/qm-dsp/hmm/hmm.h')
-rw-r--r--libs/qm-dsp/hmm/hmm.h52
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
+