diff --git a/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/Booster.scala b/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/Booster.scala index f0e01062e..760aee828 100644 --- a/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/Booster.scala +++ b/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/Booster.scala @@ -166,9 +166,9 @@ class Booster private[xgboost4j](private var booster: JBooster) * Controls whether the split statistics are output. */ @throws(classOf[XGBoostError]) - def getModelDump(featureMap: String = null, withStats: Boolean = false) + def getModelDump(featureMap: String = null, withStats: Boolean = false, format: String = "text") : Array[String] = { - booster.getModelDump(featureMap, withStats) + booster.getModelDump(featureMap, withStats, format) } /**