From 53108ed6ad241765757c1e4c68189035505b370f Mon Sep 17 00:00:00 2001 From: Manish Amde Date: Sun, 19 Jan 2014 16:56:15 -0800 Subject: [PATCH] fixing index for highest bin Signed-off-by: Manish Amde --- .../main/scala/org/apache/spark/mllib/tree/DecisionTree.scala | 2 +- 1 file changed, 1 insertion(+), 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 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)