fixing index for highest bin

Signed-off-by: Manish Amde <[email protected]>
witgo · Feb 28, 2014 · 53108ed · 53108ed
1 parent e23c2e5
commit 53108ed
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -629,7 +629,7 @@ object DecisionTree extends Serializable with Logging {
             bins(featureIndex)(index) = bin
           }
           bins(featureIndex)(numBins-1)
-            = new Bin(splits(featureIndex)(numBins-3),new DummyHighSplit(Continuous),Continuous)
+            = new Bin(splits(featureIndex)(numBins-2),new DummyHighSplit(Continuous),Continuous)
         }
 
         (splits,bins)