Skip to content

Commit ce004a1

Browse files
committed
minor formatting
1 parent b27ad2c commit ce004a1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)