fix seed in tests

apache · Apr 21, 2015 · 89321ee · 89321ee
1 parent 3ad2387
commit 89321ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py
@@ -310,7 +310,7 @@ def test_classification(self):
 
         rf_model = RandomForest.trainClassifier(
             rdd, numClasses=2, categoricalFeaturesInfo=categoricalFeaturesInfo, numTrees=10,
-            maxBins=4)
+            maxBins=4, seed=1)
         self.assertTrue(rf_model.predict(features[0]) <= 0)
         self.assertTrue(rf_model.predict(features[1]) > 0)
         self.assertTrue(rf_model.predict(features[2]) <= 0)