diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala index c01b31140..6e980f8aa 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/BoosterParams.scala @@ -154,7 +154,7 @@ trait BoosterParams extends Params { /** * Control the balance of positive and negative weights, useful for unbalanced classes. A typical - * value to consider: sum(negative cases) / sum(positive cases). [default=0] + * value to consider: sum(negative cases) / sum(positive cases). [default=1] */ val scalePosWeight = new DoubleParam(this, "scale_pos_weight", "Control the balance of positive" + " and negative weights, useful for unbalanced classes. A typical value to consider:" + diff --git a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSparkPipelinePersistence.scala b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSparkPipelinePersistence.scala index 031a68271..f984a9da3 100644 --- a/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSparkPipelinePersistence.scala +++ b/jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostSparkPipelinePersistence.scala @@ -90,7 +90,7 @@ class XGBoostSparkPipelinePersistence extends FunSuite with PerTest .setInputCols(df.columns .filter(!_.contains("label"))) .setOutputCol("features") - val xgbEstimator = new XGBoostEstimator(Map("num_rounds" -> 10, + val xgbEstimator = new XGBoostEstimator(Map("num_round" -> 10, "tracker_conf" -> TrackerConf(60 * 60 * 1000, "scala") )).setFeaturesCol("features").setLabelCol("label") // separate