From d333918f5e95ce0af7a71993541632514cb5beba Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Sun, 7 Jul 2019 16:02:44 -0700 Subject: [PATCH] [jvm-packages] Expose setMissing method in XGBoostClassificationModel / XGBoostRegressionModel (#4643) --- .../scala/ml/dmlc/xgboost4j/scala/spark/XGBoostClassifier.scala | 2 ++ .../scala/ml/dmlc/xgboost4j/scala/spark/XGBoostRegressor.scala | 2 ++ 2 files changed, 4 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 955a2401e..832826bf8 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 @@ -247,6 +247,8 @@ class XGBoostClassificationModel private[ml]( def setTreeLimit(value: Int): this.type = set(treeLimit, value) + def setMissing(value: Float): this.type = set(missing, value) + def setInferBatchSize(value: Int): this.type = set(inferBatchSize, value) /** 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 46dd0e6ed..f447042e2 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 @@ -242,6 +242,8 @@ class XGBoostRegressionModel private[ml] ( def setTreeLimit(value: Int): this.type = set(treeLimit, value) + def setMissing(value: Float): this.type = set(missing, value) + def setInferBatchSize(value: Int): this.type = set(inferBatchSize, value) /**