@@ -60,7 +60,7 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
60
60
// depth of the decision tree
61
61
val maxDepth = strategy.maxDepth
62
62
// the max number of nodes possible given the depth of the tree
63
- val maxNumNodes = scala. math.pow(2 , maxDepth).toInt - 1
63
+ val maxNumNodes = math.pow(2 , maxDepth).toInt - 1
64
64
// Initialize an array to hold filters applied to points for each node.
65
65
val filters = new Array [List [Filter ]](maxNumNodes)
66
66
// The filter at the top node is an empty list.
@@ -85,11 +85,11 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
85
85
}
86
86
logDebug(" numElementsPerNode = " + numElementsPerNode)
87
87
val arraySizePerNode = 8 * numElementsPerNode // approx. memory usage for bin aggregate array
88
- val maxNumberOfNodesPerGroup = scala. math.max(maxMemoryUsage / arraySizePerNode, 1 )
88
+ val maxNumberOfNodesPerGroup = math.max(maxMemoryUsage / arraySizePerNode, 1 )
89
89
logDebug(" maxNumberOfNodesPerGroup = " + maxNumberOfNodesPerGroup)
90
90
// nodes at a level is 2^(level-1). level is zero indexed.
91
- val maxLevelForSingleGroup = scala. math.max(
92
- (scala. math.log(maxNumberOfNodesPerGroup) / scala. math.log(2 )).floor.toInt - 1 , 0 )
91
+ val maxLevelForSingleGroup = math.max(
92
+ (math.log(maxNumberOfNodesPerGroup) / math.log(2 )).floor.toInt - 1 , 0 )
93
93
logDebug(" max level for single group = " + maxLevelForSingleGroup)
94
94
95
95
/*
@@ -120,7 +120,7 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
120
120
filters)
121
121
logDebug(" final best split = " + nodeSplitStats._1)
122
122
}
123
- require(scala. math.pow(2 , level) == splitsStatsForLevel.length)
123
+ require(math.pow(2 , level) == splitsStatsForLevel.length)
124
124
// Check whether all the nodes at the current level at leaves.
125
125
val allLeaf = splitsStatsForLevel.forall(_._2.gain <= 0 )
126
126
logDebug(" all leaf = " + allLeaf)
@@ -153,7 +153,7 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
153
153
nodes : Array [Node ]): Unit = {
154
154
val split = nodeSplitStats._1
155
155
val stats = nodeSplitStats._2
156
- val nodeIndex = scala. math.pow(2 , level).toInt - 1 + index
156
+ val nodeIndex = math.pow(2 , level).toInt - 1 + index
157
157
val isLeaf = (stats.gain <= 0 ) || (level == strategy.maxDepth - 1 )
158
158
val node = new Node (nodeIndex, stats.predict, isLeaf, Some (split), None , None , Some (stats))
159
159
logDebug(" Node = " + node)
@@ -174,7 +174,7 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
174
174
var i = 0
175
175
while (i <= 1 ) {
176
176
// Calculate the index of the node from the node level and the index at the current level.
177
- val nodeIndex = scala. math.pow(2 , level + 1 ).toInt - 1 + 2 * index + i
177
+ val nodeIndex = math.pow(2 , level + 1 ).toInt - 1 + 2 * index + i
178
178
if (level < maxDepth - 1 ) {
179
179
val impurity = if (i == 0 ) {
180
180
nodeSplitStats._2.leftImpurity
@@ -300,7 +300,7 @@ object DecisionTree extends Serializable with Logging {
300
300
maxLevelForSingleGroup : Int ): Array [(Split , InformationGainStats )] = {
301
301
// split into groups to avoid memory overflow during aggregation
302
302
if (level > maxLevelForSingleGroup) {
303
- val numGroups = scala. math.pow(2 , (level - maxLevelForSingleGroup)).toInt
303
+ val numGroups = math.pow(2 , (level - maxLevelForSingleGroup)).toInt
304
304
logDebug(" numGroups = " + numGroups)
305
305
var groupIndex = 0
306
306
var bestSplits = new Array [(Split , InformationGainStats )](0 )
@@ -366,7 +366,7 @@ object DecisionTree extends Serializable with Logging {
366
366
*/
367
367
368
368
// common calculations for multiple nested methods
369
- val numNodes = scala. math.pow(2 , level).toInt / numGroups
369
+ val numNodes = math.pow(2 , level).toInt / numGroups
370
370
logDebug(" numNodes = " + numNodes)
371
371
// Find the number of features by looking at the first sample.
372
372
val numFeatures = input.first().features.size
@@ -382,7 +382,7 @@ object DecisionTree extends Serializable with Logging {
382
382
if (level == 0 ) {
383
383
List [Filter ]()
384
384
} else {
385
- val nodeFilterIndex = scala. math.pow(2 , level).toInt - 1 + nodeIndex + groupShift
385
+ val nodeFilterIndex = math.pow(2 , level).toInt - 1 + nodeIndex + groupShift
386
386
filters(nodeFilterIndex)
387
387
}
388
388
}
@@ -951,7 +951,7 @@ object DecisionTree extends Serializable with Logging {
951
951
// Iterating over all nodes at this level
952
952
var node = 0
953
953
while (node < numNodes) {
954
- val nodeImpurityIndex = scala. math.pow(2 , level).toInt - 1 + node + groupShift
954
+ val nodeImpurityIndex = math.pow(2 , level).toInt - 1 + node + groupShift
955
955
val binsForNode : Array [Double ] = getBinDataForNode(node)
956
956
logDebug(" nodeImpurityIndex = " + nodeImpurityIndex)
957
957
val parentNodeImpurity = parentImpurities(nodeImpurityIndex)
0 commit comments