Bug fix

freeman-lab · freeman-lab · commit 9facbe3ecbc1 · 2014-10-25T04:04:11.000-04: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
@@ -165,7 +165,7 @@ class StreamingKMeans(
   /** Initialize random centers, requiring only the number of dimensions. */
   def setRandomCenters(d: Int): this.type = {
     val initialCenters = (0 until k).map(_ => Vectors.dense(Array.fill(d)(nextGaussian()))).toArray
-    val clusterCounts = Array.fill(0)(d).map(_.toLong)
+    val clusterCounts = Array.fill(this.k)(0).map(_.toLong)
     this.model = new StreamingKMeansModel(initialCenters, clusterCounts)
     this
   }

Original file line number	Diff line number	Diff line change
`@@ -165,7 +165,7 @@ class StreamingKMeans(`
`165`	`165`	`/** Initialize random centers, requiring only the number of dimensions. */`
`166`	`166`	`def setRandomCenters(d: Int): this.type = {`
`167`	`167`	`val initialCenters = (0 until k).map(_ => Vectors.dense(Array.fill(d)(nextGaussian()))).toArray`
`168`		`- val clusterCounts = Array.fill(0)(d).map(_.toLong)`
	`168`	`+ val clusterCounts = Array.fill(this.k)(0).map(_.toLong)`
`169`	`169`	`this.model = new StreamingKMeansModel(initialCenters, clusterCounts)`
`170`	`170`	`this`
`171`	`171`	`}`