package ml
DataFrame-based machine learning APIs to let users quickly assemble and configure practical machine learning pipelines.
- Alphabetic
- By Inheritance
- ml
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Type Members
-
abstract
class
Estimator[M <: Model[M]] extends PipelineStage
:: DeveloperApi :: Abstract class for estimators that fit models to data.
:: DeveloperApi :: Abstract class for estimators that fit models to data.
- Annotations
- @DeveloperApi()
-
abstract
class
Model[M <: Model[M]] extends Transformer
:: DeveloperApi :: A fitted model, i.e., a Transformer produced by an Estimator.
:: DeveloperApi :: A fitted model, i.e., a Transformer produced by an Estimator.
- M
model type
- Annotations
- @DeveloperApi()
-
class
Pipeline extends Estimator[PipelineModel] with MLWritable
A simple pipeline, which acts as an estimator.
A simple pipeline, which acts as an estimator. A Pipeline consists of a sequence of stages, each of which is either an Estimator or a Transformer. When
Pipeline.fitis called, the stages are executed in order. If a stage is an Estimator, itsEstimator.fitmethod will be called on the input dataset to fit a model. Then the model, which is a transformer, will be used to transform the dataset as the input to the next stage. If a stage is a Transformer, itsTransformer.transformmethod will be called to produce the dataset for the next stage. The fitted model from a Pipeline is a PipelineModel, which consists of fitted models and transformers, corresponding to the pipeline stages. If there are no stages, the pipeline acts as an identity transformer.- Annotations
- @Since( "1.2.0" )
-
class
PipelineModel extends Model[PipelineModel] with MLWritable with Logging
Represents a fitted pipeline.
Represents a fitted pipeline.
- Annotations
- @Since( "1.2.0" )
-
abstract
class
PipelineStage extends Params with Logging
:: DeveloperApi :: A stage in a pipeline, either an Estimator or a Transformer.
:: DeveloperApi :: A stage in a pipeline, either an Estimator or a Transformer.
- Annotations
- @DeveloperApi()
-
abstract
class
PredictionModel[FeaturesType, M <: PredictionModel[FeaturesType, M]] extends Model[M] with PredictorParams
:: DeveloperApi :: Abstraction for a model for prediction tasks (regression and classification).
:: DeveloperApi :: Abstraction for a model for prediction tasks (regression and classification).
- FeaturesType
Type of features. E.g.,
VectorUDTfor vector features.- M
Specialization of PredictionModel. If you subclass this type, use this type parameter to specify the concrete type for the corresponding model.
- Annotations
- @DeveloperApi()
-
abstract
class
Predictor[FeaturesType, Learner <: Predictor[FeaturesType, Learner, M], M <: PredictionModel[FeaturesType, M]] extends Estimator[M] with PredictorParams
:: DeveloperApi :: Abstraction for prediction problems (regression and classification).
:: DeveloperApi :: Abstraction for prediction problems (regression and classification). It accepts all NumericType labels and will automatically cast it to DoubleType in
fit(). If this predictor supports weights, it accepts all NumericType weights, which will be automatically casted to DoubleType infit().- FeaturesType
Type of features. E.g.,
VectorUDTfor vector features.- Learner
Specialization of this class. If you subclass this type, use this type parameter to specify the concrete type.
- M
Specialization of PredictionModel. If you subclass this type, use this type parameter to specify the concrete type for the corresponding model.
- Annotations
- @DeveloperApi()
-
abstract
class
Transformer extends PipelineStage
:: DeveloperApi :: Abstract class for transformers that transform one dataset into another.
:: DeveloperApi :: Abstract class for transformers that transform one dataset into another.
- Annotations
- @DeveloperApi()
-
abstract
class
UnaryTransformer[IN, OUT, T <: UnaryTransformer[IN, OUT, T]] extends Transformer with HasInputCol with HasOutputCol with Logging
:: DeveloperApi :: Abstract class for transformers that take one input column, apply transformation, and output the result as a new column.
:: DeveloperApi :: Abstract class for transformers that take one input column, apply transformation, and output the result as a new column.
- Annotations
- @DeveloperApi()
Value Members
-
object
Pipeline extends MLReadable[Pipeline] with Serializable
- Annotations
- @Since( "1.6.0" )
-
object
PipelineModel extends MLReadable[PipelineModel] with Serializable
- Annotations
- @Since( "1.6.0" )