From 730000251635e57c96098273d3283080e24aa598 Mon Sep 17 00:00:00 2001 From: Matthew Tovbin Date: Fri, 10 Aug 2018 09:38:58 -0700 Subject: [PATCH] [jvm-packages] Use treeLimit param in getTreeLimit (#3575) --- .../ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala index 89e2ef102..d67d0d4e0 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala @@ -240,7 +240,7 @@ private[spark] trait BoosterParams extends Params { final val treeLimit = new IntParam(this, name = "treeLimit", doc = "number of trees used in the prediction; defaults to 0 (use all trees).") - final def getTreeLimit: Double = $(lambdaBias) + final def getTreeLimit: Double = $(treeLimit) setDefault(eta -> 0.3, gamma -> 0, maxDepth -> 6, minChildWeight -> 1, maxDeltaStep -> 0,