fix random forest unit test

apache · Oct 13, 2014 · 3c72913 · 3c72913
1 parent 092efcb
commit 3c72913
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/tree/RandomForestSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/tree/RandomForestSuite.scala
@@ -68,7 +68,8 @@ class RandomForestSuite extends FunSuite with LocalSparkContext {
     val categoricalFeaturesInfo = Map.empty[Int, Int]
     val numTrees = 1
 
-    val strategy = new Strategy(algo = Classification, impurity = Gini, maxDepth = 2,
+    val strategy = new Strategy(algo = Classification, impurity = Gini,
+      maxDepth = 2, maxBins = 10,
       numClassesForClassification = 2, categoricalFeaturesInfo = categoricalFeaturesInfo)
 
     val rf = RandomForest.trainClassifier(rdd, strategy, numTrees = numTrees,