diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala index eb184f1a2154..62d5eb0b4b71 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala @@ -282,7 +282,7 @@ object XGBoost extends Serializable { s" set value $missing) when you have SparseVector or Empty vector as your feature" + s" format. If you didn't use Spark's VectorAssembler class to build your feature " + s"vector but instead did so in a way that preserves zeros in your feature vector " + - s"you can avoid this check by using the 'allow_non_zero_missing parameter'" + + s"you can avoid this check by using the 'allow_non_zero_for_missing parameter'" + s" (only use if you know what you are doing)") } labeledPoint diff --git a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/MissingValueHandlingSuite.scala b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/MissingValueHandlingSuite.scala index 75106bec42bf..9e23d81b51d1 100644 --- a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/MissingValueHandlingSuite.scala +++ b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/MissingValueHandlingSuite.scala @@ -145,7 +145,7 @@ class MissingValueHandlingSuite extends FunSuite with PerTest { } } - test("specify a non-zero missing value but set allow_non_zero_missing " + + test("specify a non-zero missing value but set allow_non_zero_for_missing " + "does not stop application") { val spark = ss import spark.implicits._