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 1ff8c05bcb790..975dd4f0cd7e7 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 @@ -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)