We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent afced16 commit c8428c4Copy full SHA for c8428c4
mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
@@ -606,8 +606,8 @@ object DecisionTree extends Serializable with Logging {
606
val arrIndex = arrShift + featureIndex
607
// Update the left or right count for one bin.
608
val aggShift = numClasses * numBins * numFeatures * nodeIndex
609
- val aggIndex = aggShift + numClasses * featureIndex * numBins
610
- + arr(arrIndex).toInt * numClasses
+ val aggIndex
+ = aggShift + numClasses * featureIndex * numBins + arr(arrIndex).toInt * numClasses
611
val labelInt = label.toInt
612
agg(aggIndex + labelInt) = agg(aggIndex + labelInt) + 1
613
}
0 commit comments