Packages

class GaussianMixtureModel extends Serializable with Saveable

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.

Annotations
@Since( "1.3.0" )
Linear Supertypes
Saveable, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. GaussianMixtureModel
  2. Saveable
  3. Serializable
  4. Serializable
  5. AnyRef
  6. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new GaussianMixtureModel(weights: Array[Double], gaussians: Array[MultivariateGaussian])

    weights

    Weights for each Gaussian distribution in the mixture, where weights(i) is the weight for Gaussian i, and weights.sum == 1

    gaussians

    Array of MultivariateGaussian where gaussians(i) represents the Multivariate Gaussian (Normal) Distribution for Gaussian i

    Annotations
    @Since( "1.3.0" )

Value Members

  1. val gaussians: Array[MultivariateGaussian]
    Annotations
    @Since( "1.3.0" )
  2. def k: Int

    Number of gaussians in mixture

    Number of gaussians in mixture

    Annotations
    @Since( "1.3.0" )
  3. def predict(points: JavaRDD[Vector]): JavaRDD[Integer]

    Java-friendly version of predict()

    Java-friendly version of predict()

    Annotations
    @Since( "1.4.0" )
  4. def predict(point: Vector): Int

    Maps given point to its cluster index.

    Maps given point to its cluster index.

    Annotations
    @Since( "1.5.0" )
  5. def predict(points: RDD[Vector]): RDD[Int]

    Maps given points to their cluster indices.

    Maps given points to their cluster indices.

    Annotations
    @Since( "1.3.0" )
  6. def predictSoft(point: Vector): Array[Double]

    Given the input vector, return the membership values to all mixture components.

    Given the input vector, return the membership values to all mixture components.

    Annotations
    @Since( "1.4.0" )
  7. def predictSoft(points: RDD[Vector]): RDD[Array[Double]]

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

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

    Annotations
    @Since( "1.3.0" )
  8. def save(sc: SparkContext, path: String): Unit

    Save this model to the given path.

    Save this model to the given path.

    This saves:

    • human-readable (JSON) model metadata to path/metadata/
    • Parquet formatted data to path/data/

    The model may be loaded using Loader.load.

    sc

    Spark context used to save model data.

    path

    Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.

    Definition Classes
    GaussianMixtureModelSaveable
    Annotations
    @Since( "1.4.0" )
  9. val weights: Array[Double]
    Annotations
    @Since( "1.3.0" )