Simpler constructor

freeman-lab · freeman-lab · commit 44050a92a4b7 · 2014-10-28T22:13:58.000-07:00
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
   }

Original file line number	Diff line number	Diff line change
`@@ -178,7 +178,7 @@ class StreamingKMeans(`
`178`	`178`
`179`	`179`	`/** Specify initial centers directly. */`
`180`	`180`	`def setInitialCenters(initialCenters: Array[Vector]): this.type = {`
`181`		`- val clusterCounts = Array.fill(this.k)(0).map(_.toLong)`
	`181`	`+ val clusterCounts = new Array[Long](this.k)`
`182`	`182`	`this.model = new StreamingKMeansModel(initialCenters, clusterCounts)`
`183`	`183`	`this`
`184`	`184`	`}`