From aec6299c4917c07b2935b052bb421df72b6cf460 Mon Sep 17 00:00:00 2001 From: Yanbo Liang Date: Sun, 1 Jul 2018 15:06:16 -0700 Subject: [PATCH] [jvm-packages] Expose nativeBooster for XGBoostClassificationModel and XGBoostRegressionModel. (#3428) --- .../ml/dmlc/xgboost4j/scala/spark/XGBoostClassifier.scala | 6 ++++++ .../ml/dmlc/xgboost4j/scala/spark/XGBoostRegressor.scala | 6 ++++++ 2 files changed, 12 insertions(+) diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostClassifier.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostClassifier.scala index 243f2decb..dfcedd3ed 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostClassifier.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostClassifier.scala @@ -211,6 +211,12 @@ class XGBoostClassificationModel private[ml]( // only called in copy() def this(uid: String) = this(uid, 2, null) + /** + * Get the native booster instance of this model. + * This is used to call low-level APIs on native booster, such as "getFeatureScore". + */ + def nativeBooster: Booster = _booster + private var trainingSummary: Option[XGBoostTrainingSummary] = None /** diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostRegressor.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostRegressor.scala index 097a0434c..dcc6e534a 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostRegressor.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostRegressor.scala @@ -207,6 +207,12 @@ class XGBoostRegressionModel private[ml] ( // only called in copy() def this(uid: String) = this(uid, null) + /** + * Get the native booster instance of this model. + * This is used to call low-level APIs on native booster, such as "getFeatureScore". + */ + def nativeBooster: Booster = _booster + private var trainingSummary: Option[XGBoostTrainingSummary] = None /**