From ef806035dac2971920b697d7a893e27288d87ef1 Mon Sep 17 00:00:00 2001 From: johnnywalleye Date: Wed, 9 Jul 2014 09:25:37 -0400 Subject: [PATCH] [SPARK-2417][MLlib] Fix DecisionTree tests --- .../org/apache/spark/mllib/tree/DecisionTreeSuite.scala | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala index 35e92d71dc63f..bcb11876b8f4f 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala @@ -253,8 +253,8 @@ class DecisionTreeSuite extends FunSuite with LocalSparkContext { val stats = bestSplits(0)._2 assert(stats.gain > 0) - assert(stats.predict > 0.4) - assert(stats.predict < 0.5) + assert(stats.predict > 0.5) + assert(stats.predict < 0.7) assert(stats.impurity > 0.2) } @@ -280,8 +280,8 @@ class DecisionTreeSuite extends FunSuite with LocalSparkContext { val stats = bestSplits(0)._2 assert(stats.gain > 0) - assert(stats.predict > 0.4) - assert(stats.predict < 0.5) + assert(stats.predict > 0.5) + assert(stats.predict < 0.7) assert(stats.impurity > 0.2) }