[jvm-packages] Expose json dumps to scala (#2247)

* Add parameter passthru of format on Booster.getModelDump
2017-05-02 17:41:27 -07:00 · 2017-05-02 17:41:27 -07:00 · 197a9eacc5
commit 197a9eacc5
parent ccccf8a015
1 changed files with 2 additions and 2 deletions
--- 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)
  }

  /**