Set a random seed in the unit testing

rxin · Jul 1, 2015 · 220a176 · 220a176
1 parent 92c3efc
commit 220a176
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala
@@ -98,7 +98,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext {
 
   test("fit & transform") {
     val predictionColName = "kmeans_prediction"
-    val kmeans = new KMeans().setK(k).setPredictionCol(predictionColName)
+    val kmeans = new KMeans().setK(k).setPredictionCol(predictionColName).setSeed(1)
     val model = kmeans.fit(dataset)
     assert(model.clusterCenters.length === k)