Github user jkbradley commented on a diff in the pull request:
https://github.com/apache/spark/pull/21129#discussion_r187113953
--- Diff: mllib/src/main/scala/org/apache/spark/ml/tree/treeParams.scala ---
@@ -460,18 +461,29 @@ private[ml] trait RandomForestRegressorParams
*
* Note: Marked as private and DeveloperApi since this may be made public
in the future.
*/
-private[ml] trait GBTParams extends TreeEnsembleParams with HasMaxIter
with HasStepSize {
+private[ml] trait GBTParams extends TreeEnsembleParams with HasMaxIter
with HasStepSize
+ with HasValidationIndicatorCol {
- /* TODO: Add this doc when we add this param. SPARK-7132
- * Threshold for stopping early when runWithValidation is used.
+ /**
+ * Threshold for stopping early when fit with validation is used.
* If the error rate on the validation input changes by less than the
validationTol,
- * then learning will stop early (before [[numIterations]]).
- * This parameter is ignored when run is used.
+ * then learning will stop early (before [[maxIter]]).
+ * This parameter is ignored when fit without validation is used.
* (default = 1e-5)
* @group param
--- End diff --
Let's add a `@see` validationIndicatorCol
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
