regression

Type Members

1. class AFTSurvivalRegression extends Estimator[AFTSurvivalRegressionModel] with AFTSurvivalRegressionParams with DefaultParamsWritable with Logging

   

   :: Experimental :: Fit a parametric survival regression model named accelerated failure time (AFT) model (https://en.wikipedia.org/wiki/Accelerated_failure_time_model) based on the Weibull distribution of the survival time.

   :: Experimental :: Fit a parametric survival regression model named accelerated failure time (AFT) model (https://en.wikipedia.org/wiki/Accelerated_failure_time_model) based on the Weibull distribution of the survival time.

   Annotations

   @Experimental() @Since( "1.6.0" )

2. class AFTSurvivalRegressionModel extends Model[AFTSurvivalRegressionModel] with AFTSurvivalRegressionParams with MLWritable

   

   :: Experimental :: Model produced by AFTSurvivalRegression.

   :: Experimental :: Model produced by AFTSurvivalRegression.

   Annotations

   @Experimental() @Since( "1.6.0" )

3. final class DecisionTreeRegressionModel extends PredictionModel[Vector, DecisionTreeRegressionModel] with DecisionTreeModel with Serializable

   

   :: Experimental :: Decision tree model for regression.

   :: Experimental :: Decision tree model for regression. It supports both continuous and categorical features.

   Annotations

   @Since( "1.4.0" ) @Experimental()

4. final class DecisionTreeRegressor extends Predictor[Vector, DecisionTreeRegressor, DecisionTreeRegressionModel] with DecisionTreeParams with TreeRegressorParams

   

   :: Experimental :: Decision tree learning algorithm for regression.

   :: Experimental :: Decision tree learning algorithm for regression. It supports both continuous and categorical features.

   Annotations

   @Since( "1.4.0" ) @Experimental()

5. final class GBTRegressionModel extends PredictionModel[Vector, GBTRegressionModel] with TreeEnsembleModel with Serializable

   

   :: Experimental ::

   :: Experimental ::

   Gradient-Boosted Trees (GBTs) model for regression. It supports both continuous and categorical features.

   Annotations

   @Since( "1.4.0" ) @Experimental()

6. final class GBTRegressor extends Predictor[Vector, GBTRegressor, GBTRegressionModel] with GBTParams with TreeRegressorParams with Logging

   

   :: Experimental :: Gradient-Boosted Trees (GBTs) learning algorithm for regression.

   :: Experimental :: Gradient-Boosted Trees (GBTs) learning algorithm for regression. It supports both continuous and categorical features.

   Annotations

   @Since( "1.4.0" ) @Experimental()

7. class IsotonicRegression extends Estimator[IsotonicRegressionModel] with IsotonicRegressionBase with DefaultParamsWritable

   

   :: Experimental :: Isotonic regression.

   :: Experimental :: Isotonic regression.

   Currently implemented using parallelized pool adjacent violators algorithm. Only univariate (single feature) algorithm supported.

   Uses org.apache.spark.mllib.regression.IsotonicRegression.

   Annotations

   @Since( "1.5.0" ) @Experimental()

8. class IsotonicRegressionModel extends Model[IsotonicRegressionModel] with IsotonicRegressionBase with MLWritable

   

   :: Experimental :: Model fitted by IsotonicRegression.

   :: Experimental :: Model fitted by IsotonicRegression. Predicts using a piecewise linear function.

   For detailed rules see org.apache.spark.mllib.regression.IsotonicRegressionModel.predict().

   Annotations

   @Since( "1.5.0" ) @Experimental()

9. class LinearRegression extends Regressor[Vector, LinearRegression, LinearRegressionModel] with LinearRegressionParams with DefaultParamsWritable with Logging

   

   :: Experimental :: Linear regression.

   :: Experimental :: Linear regression.

   The learning objective is to minimize the squared error, with regularization. The specific squared error loss function used is: L = 1/2n ||A coefficients - y||²

   This support multiple types of regularization:

   • none (a.k.a. ordinary least squares)
   • L2 (ridge regression)
   • L1 (Lasso)
   • L2 + L1 (elastic net)

   Annotations

   @Since( "1.3.0" ) @Experimental()

10. class LinearRegressionModel extends RegressionModel[Vector, LinearRegressionModel] with LinearRegressionParams with MLWritable

    

    :: Experimental :: Model produced by LinearRegression.

    :: Experimental :: Model produced by LinearRegression.

    Annotations

    @Since( "1.3.0" ) @Experimental()

11. class LinearRegressionSummary extends Serializable

    

    :: Experimental :: Linear regression results evaluated on a dataset.

    :: Experimental :: Linear regression results evaluated on a dataset.

    Annotations

    @Since( "1.5.0" ) @Experimental()

12. class LinearRegressionTrainingSummary extends LinearRegressionSummary

    

    :: Experimental :: Linear regression training results.

    :: Experimental :: Linear regression training results. Currently, the training summary ignores the training coefficients except for the objective trace.

    Annotations

    @Since( "1.5.0" ) @Experimental()

13. final class RandomForestRegressionModel extends PredictionModel[Vector, RandomForestRegressionModel] with TreeEnsembleModel with Serializable

    

    :: Experimental :: Random Forest model for regression.

    :: Experimental :: Random Forest model for regression. It supports both continuous and categorical features.

    Annotations

    @Since( "1.4.0" ) @Experimental()

14. final class RandomForestRegressor extends Predictor[Vector, RandomForestRegressor, RandomForestRegressionModel] with RandomForestParams with TreeRegressorParams

    

    :: Experimental :: Random Forest learning algorithm for regression.

    :: Experimental :: Random Forest learning algorithm for regression. It supports both continuous and categorical features.

    Annotations

    @Since( "1.4.0" ) @Experimental()

15. abstract class RegressionModel[FeaturesType, M <: RegressionModel[FeaturesType, M]] extends PredictionModel[FeaturesType, M] with PredictorParams

    

    :: DeveloperApi ::

    :: DeveloperApi ::

    Model produced by a Regressor.

    FeaturesType

    Type of input features. E.g., org.apache.spark.mllib.linalg.Vector

    M

    Concrete Model type.

    Annotations

    @DeveloperApi()