Merge pull request #14 from mengxr/SPARK-6953

fix seed in tests
apache · Apr 21, 2015 · d08542d · d08542d
2 parents 3ad2387 + 89321ee
commit d08542d
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)