File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
mllib/src/main/scala/org/apache/spark/mllib/tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -77,12 +77,12 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
77
77
// Max memory usage for aggregates
78
78
val maxMemoryUsage = strategy.maxMemoryInMB * 1024 * 1024
79
79
logDebug(" max memory usage for aggregates = " + maxMemoryUsage + " bytes." )
80
- val numElementsPerNode = {
80
+ val numElementsPerNode =
81
81
strategy.algo match {
82
82
case Classification => 2 * numBins * numFeatures
83
83
case Regression => 3 * numBins * numFeatures
84
84
}
85
- }
85
+
86
86
logDebug(" numElementsPerNode = " + numElementsPerNode)
87
87
val arraySizePerNode = 8 * numElementsPerNode // approx. memory usage for bin aggregate array
88
88
val maxNumberOfNodesPerGroup = math.max(maxMemoryUsage / arraySizePerNode, 1 )
You can’t perform that action at this time.
0 commit comments