From e0426ee74d5e233c1e7b14e29135015d09a0370c Mon Sep 17 00:00:00 2001 From: Manish Amde Date: Wed, 30 Apr 2014 17:36:47 -0700 Subject: [PATCH] renamed parameter --- .../main/scala/org/apache/spark/mllib/tree/DecisionTree.scala | 2 +- .../org/apache/spark/mllib/tree/configuration/Strategy.scala | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala index 6f1f3883a7e81..4af6a827946bd 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala @@ -77,7 +77,7 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo // Calculate level for single group construction // Max memory usage for aggregates - val maxMemoryUsage = strategy.maxMemory * 1024 * 1024 + val maxMemoryUsage = strategy.maxMemoryInMB * 1024 * 1024 logDebug("max memory usage for aggregates = " + maxMemoryUsage) val numElementsPerNode = { strategy.algo match { diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala index 18918ad5c746e..eeec2f1621cdd 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/tree/configuration/Strategy.scala @@ -35,7 +35,7 @@ import org.apache.spark.mllib.tree.configuration.QuantileStrategy._ * k) implies the feature n is categorical with k categories 0, * 1, 2, ... , k-1. It's important to note that features are * zero-indexed. - * @param maxMemory maximum memory in MB allocated to histogram aggregation. Default value is + * @param maxMemoryInMB maximum memory in MB allocated to histogram aggregation. Default value is * 128 MB. * */ @@ -47,4 +47,4 @@ class Strategy ( val maxBins: Int = 100, val quantileCalculationStrategy: QuantileStrategy = Sort, val categoricalFeaturesInfo: Map[Int, Int] = Map[Int, Int](), - val maxMemory: Int = 128) extends Serializable + val maxMemoryInMB: Int = 128) extends Serializable