|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |
Object org.apache.spark.ml.PipelineStage org.apache.spark.ml.Estimator<M> org.apache.spark.ml.Predictor<FeaturesType,Learner,M>
public abstract class Predictor<FeaturesType,Learner extends Predictor<FeaturesType,Learner,M>,M extends PredictionModel<FeaturesType,M>>
:: DeveloperApi :: Abstraction for prediction problems (regression and classification).
Constructor Summary | |
---|---|
Predictor()
|
Method Summary | |
---|---|
abstract Learner |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params. |
M |
fit(DataFrame dataset)
Fits a model to the input data. |
Learner |
setFeaturesCol(String value)
|
Learner |
setLabelCol(String value)
|
Learner |
setPredictionCol(String value)
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi :: |
StructType |
validateAndTransformSchema(StructType schema,
boolean fitting,
DataType featuresDataType)
Validates and transforms the input schema with the provided param map. |
Methods inherited from class org.apache.spark.ml.Estimator |
---|
fit, 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.param.Params |
---|
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams |
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 |
---|
public Predictor()
Method Detail |
---|
public Learner setLabelCol(String value)
public Learner setFeaturesCol(String value)
public Learner setPredictionCol(String value)
public M fit(DataFrame dataset)
Estimator
fit
in class Estimator<M extends PredictionModel<FeaturesType,M>>
dataset
- (undocumented)
public abstract Learner copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Estimator<M extends PredictionModel<FeaturesType,M>>
extra
- (undocumented)
defaultCopy()
public StructType transformSchema(StructType schema)
PipelineStage
Derives the output schema from the input schema.
transformSchema
in class PipelineStage
schema
- (undocumented)
public StructType validateAndTransformSchema(StructType schema, boolean fitting, DataType featuresDataType)
schema
- input schemafitting
- whether this is in fittingfeaturesDataType
- SQL DataType for FeaturesType.
E.g., VectorUDT
for vector features.
|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |