org.apache.spark.mllib.clustering

Class GaussianMixtureModel

Object

org.apache.spark.mllib.clustering.GaussianMixtureModel

All Implemented Interfaces:

java.io.Serializable

public class GaussianMixtureModel
extends Object
implements scala.Serializable

:: Experimental ::

Multivariate Gaussian Mixture Model (GMM) consisting of k Gaussians, where points are drawn from each Gaussian i=1..k with probability w(i); mu(i) and sigma(i) are the respective mean and covariance for each Gaussian distribution i=1..k.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GaussianMixtureModel(double[] weights, MultivariateGaussian[] gaussians)

Method Summary

Methods

Modifier and Type	Method and Description
MultivariateGaussian[]	gaussians()
int	k() Number of gaussians in mixture
RDD<Object>	predict(RDD<Vector> points) Maps given points to their cluster indices.
RDD<double[]>	predictSoft(RDD<Vector> points) Given the input vectors, return the membership value of each vector to all mixture components.
double[]	weights()

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

GaussianMixtureModel

public GaussianMixtureModel(double[] weights,
                    MultivariateGaussian[] gaussians)

Method Detail

weights

public double[] weights()

gaussians

public MultivariateGaussian[] gaussians()

k

public int k()

Number of gaussians in mixture

predict

public RDD<Object> predict(RDD<Vector> points)

Maps given points to their cluster indices.

predictSoft

public RDD<double[]> predictSoft(RDD<Vector> points)

Given the input vectors, return the membership value of each vector to all mixture components.