org.apache.spark.ml.impl.estimator

Class Predictor<FeaturesType,Learner extends Predictor<FeaturesType,Learner,M>,M extends PredictionModel<FeaturesType,M>>

Object

org.apache.spark.ml.PipelineStage

org.apache.spark.ml.Estimator<M>

org.apache.spark.ml.impl.estimator.Predictor<FeaturesType,Learner,M>

All Implemented Interfaces:

java.io.Serializable, Logging, Identifiable, PredictorParams, HasFeaturesCol, HasLabelCol, HasPredictionCol, Params

Direct Known Subclasses:

Classifier, Regressor

public abstract class Predictor<FeaturesType,Learner extends Predictor<FeaturesType,Learner,M>,M extends PredictionModel<FeaturesType,M>>
extends Estimator<M>
implements PredictorParams

:: AlphaComponent ::

Abstraction for prediction problems (regression and classification).

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
Predictor()

Method Summary

Methods

Modifier and Type	Method and Description
M	fit(DataFrame dataset, ParamMap paramMap) Fits a single model to the input data with provided parameter map.
Learner	setFeaturesCol(String value)
Learner	setLabelCol(String value)
Learner	setPredictionCol(String value)
org.apache.spark.sql.types.StructType	transformSchema(org.apache.spark.sql.types.StructType schema, ParamMap paramMap) :: DeveloperAPI ::

Methods inherited from class org.apache.spark.ml.Estimator

fit, fit, fit

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.spark.ml.impl.estimator.PredictorParams

validateAndTransformSchema

Methods inherited from interface org.apache.spark.ml.param.HasLabelCol

getLabelCol, labelCol

Methods inherited from interface org.apache.spark.ml.param.HasFeaturesCol

featuresCol, getFeaturesCol

Methods inherited from interface org.apache.spark.ml.param.HasPredictionCol

getPredictionCol, predictionCol

Methods inherited from interface org.apache.spark.ml.param.Params

addOutputColumn, checkInputColumn, explainParams, get, getParam, isSet, paramMap, params, set, set, validate, validate

Methods inherited from interface org.apache.spark.ml.Identifiable

uid

Methods inherited from interface org.apache.spark.Logging

initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning

Constructor Detail

Predictor

public Predictor()

Method Detail

setLabelCol

public Learner setLabelCol(String value)

setFeaturesCol

public Learner setFeaturesCol(String value)

setPredictionCol

public Learner setPredictionCol(String value)

fit

public M fit(DataFrame dataset,
    ParamMap paramMap)

Description copied from class: Estimator

Fits a single model to the input data with provided parameter map.

Specified by:

fit in class Estimator<M extends PredictionModel<FeaturesType,M>>

Parameters:

dataset - input dataset

paramMap - Parameter map. These values override any specified in this Estimator's embedded ParamMap.

Returns:

fitted model

transformSchema

public org.apache.spark.sql.types.StructType transformSchema(org.apache.spark.sql.types.StructType schema,
                                                    ParamMap paramMap)

Description copied from class: PipelineStage

:: DeveloperAPI ::

Derives the output schema from the input schema and parameters. The schema describes the columns and types of the data.

Specified by:

transformSchema in class PipelineStage

Parameters:

schema - Input schema to this stage

paramMap - Parameters passed to this stage

Returns:

Output schema from this stage