zhengruifeng created SPARK-28514:
------------------------------------

             Summary: Remove the redundant transformImpl method in RF & GBT
                 Key: SPARK-28514
                 URL: https://issues.apache.org/jira/browse/SPARK-28514
             Project: Spark
          Issue Type: Improvement
          Components: ML
    Affects Versions: 3.0.0
            Reporter: zhengruifeng


1, In GBTClassifier & RandomForestClassifier, the real transform methods 
inherit from ProbabilisticClassificationModel which can deal with multi output 
columns.

The transformImpl method, which deals with only one column - predictionCol, 
completely does nothing. This is quite confusing.

 

2, In GBTRegressor & RandomForestRegressor, the transformImpl do exactly what 
the superclass PredictionModel does (except model broadcasting), so can be 
removed.

 

 



--
This message was sent by Atlassian JIRA
(v7.6.14#76016)

---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to