Github user holdenk commented on a diff in the pull request:
https://github.com/apache/spark/pull/20701#discussion_r173550702
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/clustering/BisectingKMeansSuite.scala
---
@@ -97,6 +97,7 @@ class BisectingKMeansSuite
test("fit, transform and summary") {
val predictionColName = "bisecting_kmeans_prediction"
val bkm = new
BisectingKMeans().setK(k).setPredictionCol(predictionColName).setSeed(1)
+ .setMaxIter(2)
--- End diff --
So I'd be more comfortable having this in a separate test, 2 iterations is
not a lot.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
