Skip to content

Commit 53108ed

Browse files
committed
fixing index for highest bin
Signed-off-by: Manish Amde <manish9ue@gmail.com>
1 parent e23c2e5 commit 53108ed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -629,7 +629,7 @@ object DecisionTree extends Serializable with Logging {
629629
bins(featureIndex)(index) = bin
630630
}
631631
bins(featureIndex)(numBins-1)
632-
= new Bin(splits(featureIndex)(numBins-3),new DummyHighSplit(Continuous),Continuous)
632+
= new Bin(splits(featureIndex)(numBins-2),new DummyHighSplit(Continuous),Continuous)
633633
}
634634

635635
(splits,bins)

0 commit comments

Comments
 (0)