Simpler constructor

YanTangZhai · Oct 29, 2014 · 44050a9 · 44050a9
1 parent c7050d5
commit 44050a9
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
@@ -178,7 +178,7 @@ class StreamingKMeans(
 
   /** Specify initial centers directly. */
   def setInitialCenters(initialCenters: Array[Vector]): this.type = {
-    val clusterCounts = Array.fill(this.k)(0).map(_.toLong)
+    val clusterCounts = new Array[Long](this.k)
     this.model = new StreamingKMeansModel(initialCenters, clusterCounts)
     this
   }