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

Class SparkKMeansOperator.Model

java.lang.Object
org.apache.wayang.spark.operators.ml.SparkKMeansOperator.Model
All Implemented Interfaces:
Serializable, KMeansModel, Model, SparkMLModel<double[],Integer>
Enclosing class:
SparkKMeansOperator
public static class SparkKMeansOperator.Model extends Object implements KMeansModel, SparkMLModel<double[],Integer>
See Also:

  • Constructor Details

    • Model

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

  • Method Details

    • 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[],Integer>> transform(org.apache.spark.api.java.JavaRDD<double[]> input)
      Specified by:
      transform in interface SparkMLModel<double[],Integer>

    • predict

      public org.apache.spark.api.java.JavaRDD<Integer> predict(org.apache.spark.api.java.JavaRDD<double[]> input)
      Specified by:
      predict in interface SparkMLModel<double[],Integer>