Use collect

msjgriffiths · Oct 29, 2014 · a4a316b · a4a316b
1 parent 1472ec5
commit a4a316b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/clustering/StreamingKMeans.scala b/mllib/src/main/scala/org/apache/spark/mllib/clustering/StreamingKMeans.scala
@@ -83,7 +83,7 @@ class StreamingKMeansModel(
       (p1._1 += p2._1, p1._2 + p2._2)
     }
     val pointStats: Array[(Int, (BV[Double], Long))] =
-      closest.reduceByKey{mergeContribs}.collectAsMap().toArray
+      closest.reduceByKey(mergeContribs).collect()
 
     // implement update rule
     for (newP <- pointStats) {