[jvm-packages] Add setNumEarlyStoppingRounds (#11571)

Author: wbo4958

jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/params/XGBoostParams.scala

@@ -122,7 +122,7 @@ private[spark] trait SparkParams[T <: Params] extends HasFeaturesCols with HasFe
   final val numWorkers = new IntParam(this, "numWorkers", "Number of workers used to train xgboost",
     ParamValidators.gtEq(1))
 
-  final def getNumRound: Int = $(numRound)
+  final def getNumWorkers: Int = $(numWorkers)
 
   final val forceRepartition = new BooleanParam(this, "forceRepartition", "If the partition " +
     "is equal to numWorkers, xgboost won't repartition the dataset. Set forceRepartition to " +
@@ -133,6 +133,8 @@ private[spark] trait SparkParams[T <: Params] extends HasFeaturesCols with HasFe
   final val numRound = new IntParam(this, "numRound", "The number of rounds for boosting",
     ParamValidators.gtEq(1))
 
+  final def getNumRound: Int = $(numRound)
+
   final val numEarlyStoppingRounds = new IntParam(this, "numEarlyStoppingRounds", "Stop training " +
     "Number of rounds of decreasing eval metric to tolerate before stopping training",
     ParamValidators.gtEq(0))
@@ -216,14 +218,14 @@ private[spark] trait SparkParams[T <: Params] extends HasFeaturesCols with HasFe
     labelCol, baseMarginCol, weightCol, predictionCol, leafPredictionCol, contribPredictionCol,
     forceRepartition, featuresCols, customEval, customObj, featureTypes, featureNames)
 
-  final def getNumWorkers: Int = $(numWorkers)
-
   def setNumWorkers(value: Int): T = set(numWorkers, value).asInstanceOf[T]
 
   def setForceRepartition(value: Boolean): T = set(forceRepartition, value).asInstanceOf[T]
 
   def setNumRound(value: Int): T = set(numRound, value).asInstanceOf[T]
 
+  def setNumEarlyStoppingRounds(value: Int): T = set(numEarlyStoppingRounds, value).asInstanceOf[T]
+
   def setFeaturesCol(value: Array[String]): T = set(featuresCols, value).asInstanceOf[T]
 
   def setBaseMarginCol(value: String): T = set(baseMarginCol, value).asInstanceOf[T]

jvm-packages/xgboost4j-spark/src/test/scala/ml/dmlc/xgboost4j/scala/spark/XGBoostParamsSuite.scala

@@ -24,7 +24,6 @@ import org.scalatest.funsuite.AnyFunSuite
 class XGBoostParamsSuite extends AnyFunSuite with PerTest with TmpFolderPerSuite {
 
   test("invalid parameters") {
-    val df = smallBinaryClassificationVector
     val estimator = new XGBoostClassifier()
 
     // We didn't set it by default
@@ -52,4 +51,11 @@ class XGBoostParamsSuite extends AnyFunSuite with PerTest with TmpFolderPerSuite
     assert(v1 == 0.66f)
   }
 
+  test("setNumEarlyStoppingRounds") {
+    val estimator = new XGBoostClassifier()
+    assert(estimator.getNumEarlyStoppingRounds == 0)
+    estimator.setNumEarlyStoppingRounds(10)
+    assert(estimator.getNumEarlyStoppingRounds == 10)
+  }
+
 }