Skip to content

Commit aa4e4df

Browse files
committed
Updated DTStatsAggregator with bug fix (nodeString should not be multiplied by statsSize)
1 parent 4651154 commit aa4e4df

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/impl/DTStatsAggregator.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ private[tree] class DTStatsAggregator(
7878
/**
7979
* Number of elements for each node, corresponding to stride between nodes in [[allStats]].
8080
*/
81-
private val nodeStride: Int = featureOffsets.last * statsSize
81+
private val nodeStride: Int = featureOffsets.last
8282

8383
/**
8484
* Total number of elements stored in this aggregator.

0 commit comments

Comments
 (0)