Github user jkbradley commented on a diff in the pull request:
https://github.com/apache/spark/pull/21097#discussion_r186565756
--- Diff:
mllib/src/main/scala/org/apache/spark/ml/regression/GBTRegressor.scala ---
@@ -269,6 +269,20 @@ class GBTRegressionModel private[ml](
new OldGBTModel(OldAlgo.Regression, _trees.map(_.toOld), _treeWeights)
}
+ /**
+ * Method to compute error or loss for every iteration of gradient
boosting.
+ *
+ * @param dataset Dataset for validation.
--- End diff --
Add doc for "loss" arg, including what the options are
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
