Github user jkbradley commented on a diff in the pull request:
https://github.com/apache/spark/pull/21129#discussion_r187114172
--- 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,
--- End diff --
Let's add the more precise description from the old BoostingStrategy for
this Param.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
