コード例 #1
0
ファイル: KMeans.java プロジェクト: phamthaithinh/carbon-ml
 /**
  * This method applies a kmeans model to a given dataset
  *
  * @param kMeansModel KMeans model
  * @param data JavaRDD containing feature vectors
  * @return JavaRDD containing cluster centers
  */
 public JavaRDD<Integer> test(KMeansModel kMeansModel, JavaRDD<Vector> data) {
   return kMeansModel.predict(data);
 }
コード例 #2
0
ファイル: KMeansMP.java プロジェクト: pashadow/cloudapp-mp5
 public Tuple2<Integer, String> call(Tuple2<String, Vector> args) {
   String title = args._1();
   Vector point = args._2();
   int cluster = model.predict(point);
   return new Tuple2<Integer, String>(cluster, title);
 }