org.apache.spark.mllib.clustering

Class KMeansModel

Object

org.apache.spark.mllib.clustering.KMeansModel

All Implemented Interfaces:

java.io.Serializable, PMMLExportable, Saveable

Direct Known Subclasses:

StreamingKMeansModel

public class KMeansModel
extends Object
implements Saveable, scala.Serializable, PMMLExportable

A clustering model for K-means. Each point belongs to the cluster with the closest center.

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	KMeansModel.SaveLoadV1_0$

Constructor Summary

Constructors

Constructor and Description
KMeansModel(Iterable<Vector> centers) A Java-friendly constructor that takes an Iterable of Vectors.
KMeansModel(Vector[] clusterCenters)

Method Summary

Modifier and Type	Method and Description
Vector[]	clusterCenters()
double	computeCost(RDD<Vector> data) Return the K-means cost (sum of squared distances of points to their nearest center) for this model on the given data.
int	k() Total number of clusters.
static KMeansModel	load(SparkContext sc, String path)
JavaRDD<Integer>	predict(JavaRDD<Vector> points) Maps given points to their cluster indices.
RDD<Object>	predict(RDD<Vector> points) Maps given points to their cluster indices.
int	predict(Vector point) Returns the cluster index that a given point belongs to.
void	save(SparkContext sc, String path) Save this model to the given path.
static String	toPMML()
static void	toPMML(java.io.OutputStream outputStream)
static void	toPMML(SparkContext sc, String path)
static void	toPMML(String localPath)

Methods inherited from class Object

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

Methods inherited from interface org.apache.spark.mllib.pmml.PMMLExportable

toPMML, toPMML, toPMML, toPMML, toPMML

Constructor Detail

KMeansModel

public KMeansModel(Vector[] clusterCenters)

KMeansModel

public KMeansModel(Iterable<Vector> centers)

A Java-friendly constructor that takes an Iterable of Vectors.

Parameters:

centers - (undocumented)

Method Detail

load

public static KMeansModel load(SparkContext sc,
                               String path)

toPMML

public static void toPMML(String localPath)

toPMML

public static void toPMML(SparkContext sc,
                          String path)

toPMML

public static void toPMML(java.io.OutputStream outputStream)

toPMML

public static String toPMML()

clusterCenters

public Vector[] clusterCenters()

k

public int k()

Total number of clusters.

Returns:

(undocumented)

predict

public int predict(Vector point)

Returns the cluster index that a given point belongs to.

Parameters:

point - (undocumented)

Returns:

(undocumented)

predict

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

Maps given points to their cluster indices.

Parameters:

points - (undocumented)

Returns:

(undocumented)

predict

public JavaRDD<Integer> predict(JavaRDD<Vector> points)

Maps given points to their cluster indices.

Parameters:

points - (undocumented)

Returns:

(undocumented)

computeCost

public double computeCost(RDD<Vector> data)

Return the K-means cost (sum of squared distances of points to their nearest center) for this model on the given data.

Parameters:

data - (undocumented)

Returns:

(undocumented)

save

public void save(SparkContext sc,
                 String path)

Description copied from interface: Saveable

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.

Specified by:

save in interface Saveable

Parameters:

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.