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 870208982c36..6ec588d4bb40 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 @@ -62,7 +62,7 @@ private[spark] trait BoosterParams extends Params { "Maximum number of nodes to be added. Only relevant when grow_policy=lossguide is set.", (value: Int) => value >= 0) - final def getMaxLeaves: Int = $(maxDepth) + final def getMaxLeaves: Int = $(maxLeaves) /**