gmm.h

Go to the documentation of this file.

/*
 * This work is licensed under a Creative Commons 
 * Attribution-Noncommercial-Share Alike 3.0 United States License.
 * 
 *    http://creativecommons.org/licenses/by-nc-sa/3.0/us/
 * 
 * You are free:
 * 
 *    to Share - to copy, distribute, display, and perform the work
 *    to Remix - to make derivative works
 * 
 * Under the following conditions:
 * 
 *    Attribution. You must attribute the work in the manner specified by the
 *    author or licensor (but not in any way that suggests that they endorse you
 *    or your use of the work).
 * 
 *    Noncommercial. You may not use this work for commercial purposes.
 * 
 *    Share Alike. If you alter, transform, or build upon this work, you may
 *    distribute the resulting work only under the same or similar license to
 *    this one.
 * 
 * For any reuse or distribution, you must make clear to others the license
 * terms of this work. The best way to do this is by including this header.
 * 
 * Any of the above conditions can be waived if you get permission from the
 * copyright holder.
 * 
 * Apart from the remix rights granted under this license, nothing in this
 * license impairs or restricts the author's moral rights.
 */


#ifndef GMM_H
#define GMM_H


#include <jwsc/config.h>

#include <stdlib.h>
#include <inttypes.h>

#include "jwsc/vector/vector.h"
#include "jwsc/matrix/matrix.h"
#include "jwsc/matblock/matblock.h"


#ifdef __cplusplus
namespace jwsc {
extern "C" {
#endif


double gmm_log_likelihood_d
(
    const Matrix_d*   mu,
    const Matblock_d* sigma,
    const Vector_d*   pi,
    const Matrix_d*   x
);


void gmm_kmeans_init_d
(
    Matrix_d**      mu_out,
    Matblock_d**    sigma_out,
    Vector_d**      pi_out,
    const Matrix_d* x,
    uint32_t        K
);


void gmm_stats_init_d
(
    Matrix_d**      mu_out,
    Matblock_d**    sigma_out,
    Vector_d**      pi_out,
    const Matrix_d* x,
    uint32_t        K
);


void train_gmm_d
(
    Matrix_d**      mu_out,
    Matblock_d**    sigma_out,
    Vector_d**      pi_out,
    Matrix_d**      gamma_out,
    const Matrix_d* x,
    uint32_t        K
);


#ifdef __cplusplus
}
}
#endif


#endif