From b53a5a262c74987d85c54fd3d41bc6d28e5e1d60 Mon Sep 17 00:00:00 2001 From: Matthew Tovbin Date: Fri, 17 Aug 2018 09:36:00 -0700 Subject: [PATCH] [jvm-packages] getTreeLimit return type should be Int --- .../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 d67d0d4e0..04b2d700c 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 = $(treeLimit) + final def getTreeLimit: Int = $(treeLimit) setDefault(eta -> 0.3, gamma -> 0, maxDepth -> 6, minChildWeight -> 1, maxDeltaStep -> 0,