change the bug for logging clustersizes

apache · May 3, 2018 · 9053281 · 9053281
1 parent 07c20a4
commit 9053281
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/BisectingKMeans.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/BisectingKMeans.scala
@@ -279,7 +279,7 @@ class BisectingKMeans @Since("2.0.0") (
     val summary = new BisectingKMeansSummary(
       model.transform(dataset), $(predictionCol), $(featuresCol), $(k))
     model.setSummary(Some(summary))
-    instr.logNamedValue("clusterSizes", summary.clusterSizes.toString)
+    instr.logNamedValue("clusterSizes", summary.clusterSizes.mkString(", "))
     instr.logSuccess(model)
     model
   }

diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/GaussianMixture.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/GaussianMixture.scala
@@ -424,7 +424,7 @@ class GaussianMixture @Since("2.0.0") (
       $(predictionCol), $(probabilityCol), $(featuresCol), $(k), logLikelihood)
     model.setSummary(Some(summary))
     instr.logNamedValue("logLikelihood", logLikelihood)
-    instr.logNamedValue("clusterSizes", summary.clusterSizes.toString)
+    instr.logNamedValue("clusterSizes", summary.clusterSizes.mkString(", "))
     instr.logSuccess(model)
     model
   }

diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/KMeans.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/KMeans.scala
@@ -378,7 +378,7 @@ class KMeans @Since("1.5.0") (
       model.transform(dataset), $(predictionCol), $(featuresCol), $(k))
 
     model.setSummary(Some(summary))
-    instr.logNamedValue("clusterSizes", summary.clusterSizes.toString)
+    instr.logNamedValue("clusterSizes", summary.clusterSizes.mkString(", "))
     instr.logSuccess(model)
     if (handlePersistence) {
       instances.unpersist()