Remove setRun and fix PowerIterationClustering failed test.

viirya · viirya · commit 4c7623ff550b · 2015-12-31T13:28:16.000+08:00
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/clustering/PowerIterationClustering.scala b/mllib/src/main/scala/org/apache/spark/mllib/clustering/PowerIterationClustering.scala
@@ -265,7 +265,7 @@ object PowerIterationClustering extends Logging {
     GraphImpl.fromExistingRDDs(vD, graph.edges)
       .mapTriplets(
         e => e.attr / math.max(e.srcAttr, MLUtils.EPSILON),
-        TripletFields.Src)
+        TripletFields.All)
   }
 
   /**
@@ -385,7 +385,6 @@ object PowerIterationClustering extends Logging {
     val points = v.mapValues(x => Vectors.dense(x)).cache()
     val model = new KMeans()
       .setK(k)
-      .setRuns(5)
       .setSeed(0L)
       .run(points.values)
     points.mapValues(p => model.predict(p)).cache()
