fix bug

apache · Oct 9, 2014 · 3652823 · 3652823
1 parent af7cb79
commit 3652823
Showing 1 changed file with 2 additions and 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
@@ -916,8 +916,8 @@ object DecisionTree extends Serializable with Logging {
 
             val featureSamples = sampledInput.map(lp => lp.features(featureIndex)).sorted
             val featureSplits = findSplits(featureSamples, metadata.numSplits(featureIndex))
-            metadata.setNumBinForFeature(featureIndex, metadata.numSplits(featureIndex))
-            val numSplits = metadata.numSplits(featureIndex)
+            val numSplits = featureSplits.length
+            metadata.setNumBinForFeature(featureIndex, numSplits + 1)
             val numBins = metadata.numBins(featureIndex)
             logDebug("numSplits= " + numSplits)