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
7777 // Max memory usage for aggregates
7878 val maxMemoryUsage = strategy.maxMemoryInMB * 1024 * 1024
7979 logDebug(" max memory usage for aggregates = " + maxMemoryUsage + " bytes." )
80- val numElementsPerNode = {
80+ val numElementsPerNode =
8181 strategy.algo match {
8282 case Classification => 2 * numBins * numFeatures
8383 case Regression => 3 * numBins * numFeatures
8484 }
85- }
85+
8686 logDebug(" numElementsPerNode = " + numElementsPerNode)
8787 val arraySizePerNode = 8 * numElementsPerNode // approx. memory usage for bin aggregate array
8888 val maxNumberOfNodesPerGroup = math.max(maxMemoryUsage / arraySizePerNode, 1 )
You can’t perform that action at this time.
0 commit comments