numRuns variable modifications

avulanov · Jul 3, 2015 · d12336e · d12336e
1 parent 07f8554
commit d12336e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/clustering/KMeans.scala b/mllib/src/main/scala/org/apache/spark/mllib/clustering/KMeans.scala
@@ -209,12 +209,11 @@ class KMeans private (
 
     val initStartTime = System.nanoTime()
 
-    val numRuns = if (initialModel.nonEmpty) runs else 1
+    val numRuns = if (initialModel.isEmpty) 1 else runs
 
     val centers = initialModel match {
       case Some(kMeansCenters) => {
-        Array.tabulate(numRuns)(r => kMeansCenters.clusterCenters
-          .map(s => new VectorWithNorm(s, Vectors.norm(s, 2.0))))
+        Array(kMeansCenters.clusterCenters.map(s => new VectorWithNorm(s, Vectors.norm(s, 2.0))))
       }
       case None => {
         if (initializationMode == KMeans.RANDOM) {