[ https://issues.apache.org/jira/browse/SPARK-30377?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17007750#comment-17007750 ]
Huaxin Gao commented on SPARK-30377: ------------------------------------ Currently, IsotonicRegression and AFTSurvivalRegression extend Estimator directly. These two need a little more code changes. I will have separate tickets for these two, and then use this ticket to fix the rest. > Make Regressors extend abstract class Regressor > ----------------------------------------------- > > Key: SPARK-30377 > URL: https://issues.apache.org/jira/browse/SPARK-30377 > Project: Spark > Issue Type: Improvement > Components: ML > Affects Versions: 3.0.0 > Reporter: zhengruifeng > Priority: Major > > Just found that {{AFTSurvivalRegression}} , {{DecisionTreeRegressor}}, > {{FMRegressor}}, {{GBTRegressor}}, {{RandomForestRegressor}} directly extend > {{Predictor}} > > Only {{GeneralizedLinearRegression}} and {{LinearRegression now extend > Regressor.}} > > -- This message was sent by Atlassian Jira (v8.3.4#803005) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org