ml - Spark Project ML Library 2.2.0 API

abstract class Estimator[M <: Model[M]] extends PipelineStage

:: DeveloperApi :: Abstract class for estimators that fit models to data.
abstract class Model[M <: Model[M]] extends Transformer

:: DeveloperApi :: A fitted model, i.
class Pipeline extends Estimator[PipelineModel] with MLWritable

A simple pipeline, which acts as an estimator.
class PipelineModel extends Model[PipelineModel] with MLWritable with Logging

Represents a fitted pipeline.
abstract class PipelineStage extends Params with Logging

:: DeveloperApi :: A stage in a pipeline, either an Estimator or a Transformer.
abstract class PredictionModel[FeaturesType, M <: PredictionModel[FeaturesType, M]] extends Model[M] with PredictorParams

:: DeveloperApi :: Abstraction for a model for prediction tasks (regression and classification).
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).
abstract class Transformer extends PipelineStage

:: DeveloperApi :: Abstract class for transformers that transform one dataset into another.
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.

object Pipeline extends MLReadable[Pipeline] with Serializable

Annotations
@Since( "1.6.0" )
object PipelineModel extends MLReadable[PipelineModel] with Serializable

Annotations
@Since( "1.6.0" )
package attribute

The ML pipeline API uses DataFrames as ML datasets.
package classification
package clustering
package evaluation
package feature

The ml.feature package provides common feature transformers that help convert raw data or features into more suitable forms for model fitting.
package fpm
package linalg
package param
package recommendation
package regression
package source
package stat
package tree
package tuning
package util

ml