Package org.apache.wayang.spark.operators.ml

Class SparkKMeansOperator.Model

java.lang.Object

org.apache.wayang.spark.operators.ml.SparkKMeansOperator.Model

All Implemented Interfaces:

java.io.Serializable, KMeansModel, Model, SparkMLModel<double[],java.lang.Integer>

Enclosing class:

SparkKMeansOperator

public static class SparkKMeansOperator.Model
extends java.lang.Object
implements KMeansModel, SparkMLModel<double[],java.lang.Integer>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor	Description
Model(org.apache.spark.ml.clustering.KMeansModel model)

Method Summary

Modifier and Type	Method	Description
double[][]	getClusterCenters()
int	getK()
org.apache.spark.api.java.JavaRDD<java.lang.Integer>	predict(org.apache.spark.api.java.JavaRDD<double[]> input)
org.apache.spark.api.java.JavaRDD<Tuple2<double[],java.lang.Integer>>	transform(org.apache.spark.api.java.JavaRDD<double[]> input)

Methods inherited from class java.lang.Object

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

Constructor Detail

Model

public Model(org.apache.spark.ml.clustering.KMeansModel model)

Method Detail

getK

public int getK()

Specified by:

getK in interface KMeansModel

getClusterCenters

public double[][] getClusterCenters()

Specified by:

getClusterCenters in interface KMeansModel

transform

public org.apache.spark.api.java.JavaRDD<Tuple2<double[],java.lang.Integer>> transform(org.apache.spark.api.java.JavaRDD<double[]> input)

Specified by:

transform in interface SparkMLModel<double[],java.lang.Integer>

predict

public org.apache.spark.api.java.JavaRDD<java.lang.Integer> predict(org.apache.spark.api.java.JavaRDD<double[]> input)

Specified by:

predict in interface SparkMLModel<double[],java.lang.Integer>