regression

Type Members

class IsotonicRegressionCalibrator extends OpBinaryEstimatorWrapper[RealNN, RealNN, RealNN, IsotonicRegression, IsotonicRegressionModel]

Isotonic regression calibrator.

Isotonic regression calibrator.

Uses org.apache.spark.ml.regression.IsotonicRegression, which supports only univariate (single feature) models.
class OpDecisionTreeRegressionModel extends OpPredictionModel[DecisionTreeRegressionModel]

Class that takes in a spark DecisionTreeRegressionModel and wraps it into an OP model which returns a Prediction feature
class OpDecisionTreeRegressor extends OpPredictorWrapper[DecisionTreeRegressor, DecisionTreeRegressionModel] with OpDecisionTreeRegressorParams

Wrapper for spark Decision Tree Regressor org.apache.spark.ml.regression.DecisionTreeRegressor
class OpGBTRegressionModel extends OpPredictionModel[GBTRegressionModel]

Class that takes in a spark GBTRegressionModel and wraps it into an OP model which returns a Prediction feature
class OpGBTRegressor extends OpPredictorWrapper[GBTRegressor, GBTRegressionModel] with OpGBTRegressorParams

Wrapper for spark GBT Regressor org.apache.spark.ml.regression.GBTRegressor
class OpGeneralizedLinearRegression extends OpPredictorWrapper[GeneralizedLinearRegression, GeneralizedLinearRegressionModel] with OpGeneralizedLinearRegressionParams

Wrapper for spark Generalized Regression org.apache.spark.ml.regression.GeneralizedLinearRegression
class OpGeneralizedLinearRegressionModel extends OpPredictorWrapperModel[GeneralizedLinearRegressionModel]

Class that takes in a spark GeneralizedLinearRegressionModel and wraps it into an OP model which returns a Prediction feature
class OpLinearRegression extends OpPredictorWrapper[LinearRegression, LinearRegressionModel] with OpLinearRegressionParams

Wrapper around spark ml linear regression org.apache.spark.ml.regression.LinearRegression
class OpLinearRegressionModel extends OpPredictionModel[LinearRegressionModel]

Class that takes in a spark LinearRegressionModel and wraps it into an OP model which returns a Prediction feature
class OpRandomForestRegressionModel extends OpPredictionModel[RandomForestRegressionModel]

Class that takes in a spark RandomForestRegressionModel and wraps it into an OP model which returns a Prediction feature
class OpRandomForestRegressor extends OpPredictorWrapper[RandomForestRegressor, RandomForestRegressionModel] with OpRandomForestRegressorParams

Wrapper around sparj Random Forest Regressor org.apache.spark.ml.regression.RandomForestRegressor
class OpXGBoostRegressionModel extends OpPredictionModel[XGBoostRegressionModel]

Class that takes in a spark XGBoostRegressionModel and wraps it into an OP model which returns a Prediction feature
class OpXGBoostRegressor extends OpPredictorWrapper[XGBoostRegressor, XGBoostRegressionModel] with OpXGBoostRegressorParams

Wrapper around XGBoost regressor XGBoostRegressor
sealed trait RegressionModelsToTry extends EnumEntry with ModelsToTry

Enumeration of possible regression models in Model Selector

Value Members

object RegressionModelSelector extends ModelSelectorFactory with Product with Serializable

A factory for Regression Model Selector
object RegressionModelsToTry extends Enum[RegressionModelsToTry] with Serializable

package regression

Type Members

class IsotonicRegressionCalibrator extends OpBinaryEstimatorWrapper[RealNN, RealNN, RealNN, IsotonicRegression, IsotonicRegressionModel]

class OpDecisionTreeRegressionModel extends OpPredictionModel[DecisionTreeRegressionModel]

class OpDecisionTreeRegressor extends OpPredictorWrapper[DecisionTreeRegressor, DecisionTreeRegressionModel] with OpDecisionTreeRegressorParams

class OpGBTRegressionModel extends OpPredictionModel[GBTRegressionModel]

class OpGBTRegressor extends OpPredictorWrapper[GBTRegressor, GBTRegressionModel] with OpGBTRegressorParams

class OpGeneralizedLinearRegression extends OpPredictorWrapper[GeneralizedLinearRegression, GeneralizedLinearRegressionModel] with OpGeneralizedLinearRegressionParams

class OpGeneralizedLinearRegressionModel extends OpPredictorWrapperModel[GeneralizedLinearRegressionModel]

class OpLinearRegression extends OpPredictorWrapper[LinearRegression, LinearRegressionModel] with OpLinearRegressionParams

class OpLinearRegressionModel extends OpPredictionModel[LinearRegressionModel]

class OpRandomForestRegressionModel extends OpPredictionModel[RandomForestRegressionModel]

class OpRandomForestRegressor extends OpPredictorWrapper[RandomForestRegressor, RandomForestRegressionModel] with OpRandomForestRegressorParams

class OpXGBoostRegressionModel extends OpPredictionModel[XGBoostRegressionModel]

class OpXGBoostRegressor extends OpPredictorWrapper[XGBoostRegressor, XGBoostRegressionModel] with OpXGBoostRegressorParams

sealed trait RegressionModelsToTry extends EnumEntry with ModelsToTry

Value Members

object RegressionModelSelector extends ModelSelectorFactory with Product with Serializable

object RegressionModelsToTry extends Enum[RegressionModelsToTry] with Serializable

Ungrouped