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.fit
is called, the stages are executed in order. If a stage is an Estimator, itsEstimator.fit
method 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.transform
method 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.,
VectorUDT
for 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.,
VectorUDT
for 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" )