From ed5a2dffbf3e91215ef220ee8cf4594bb4898a9b Mon Sep 17 00:00:00 2001
From: Manish Amde <manish9ue@gmail.com>
Date: Mon, 12 May 2014 20:45:22 -0700
Subject: [PATCH] fixed classification requirements

---
 .../main/scala/org/apache/spark/mllib/tree/DecisionTree.scala | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
index 1745a4b09e3d4..072651dbf1732 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
@@ -1093,9 +1093,9 @@ object DecisionTree extends Serializable with Logging {
      */
     if (strategy.categoricalFeaturesInfo.size > 0) {
       val maxCategoriesForFeatures = strategy.categoricalFeaturesInfo.maxBy(_._2)._2
-      require(numBins >= maxCategoriesForFeatures)
+      require(numBins > maxCategoriesForFeatures)
       if (strategy.isMultiClassification) {
-        require(numBins > math.pow(2, maxCategoriesForFeatures.toInt) - 1)
+        require(numBins > math.pow(2, maxCategoriesForFeatures.toInt - 1) - 1)
       }
     }