Github user huaxingao commented on a diff in the pull request:
https://github.com/apache/spark/pull/21465#discussion_r235488017
--- Diff: python/pyspark/ml/classification.py ---
@@ -1176,8 +1176,8 @@ def trees(self):
@inherit_doc
class GBTClassifier(JavaEstimator, HasFeaturesCol, HasLabelCol,
HasPredictionCol, HasMaxIter,
- GBTParams, HasCheckpointInterval, HasStepSize,
HasSeed, JavaMLWritable,
- JavaMLReadable):
+ GBTParams, HasCheckpointInterval, HasStepSize, HasSeed,
+ HasValidationIndicatorCol, JavaMLWritable,
JavaMLReadable):
--- End diff --
@BryanCutler Thank you very much for reviewing my PR. I moved
HasValidationIndicatorCol, HasMaxIter and HasStepSize to GBTParams.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
