expectation-maximization
Gaussian Mixture Models using Expectation Maximization algorithm transcribed from Expectation Maximization repository of Ophir LOJKINE, also using his multivariate gaussian link.
Installation
$ npm install ml-expectation-maximization
Example
const ExpectationMaximization = ExpectationMaximization;const em = ;em; // data is a training matrixem; // data matrix to predict /*Get information about clusters obtained on the training step.Each element of the array is a cluster with the following information * weight: Weight of the current cluster.* mean: Current mean of the cluster.* covariance: Covariance matrix of the cluster.* prediction: prediction label associated with the cluster.*/var data = em; // save your modelvar model = em; // load your modelvar newEM = EM;