public abstract class Predictor<FeaturesType,Learner extends Predictor<FeaturesType,Learner,M>,M extends PredictionModel<FeaturesType,M>> extends Estimator<M>
Constructor and Description |
---|
Predictor() |
Modifier and Type | Method and Description |
---|---|
abstract Learner |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
Param<String> |
featuresCol()
Param for features column name.
|
M |
fit(DataFrame dataset)
Fits a model to the input data.
|
String |
getFeaturesCol() |
String |
getLabelCol() |
String |
getPredictionCol() |
Param<String> |
labelCol()
Param for label column name.
|
Param<String> |
predictionCol()
Param for prediction column name.
|
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.
|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn, validateParams
toString, uid
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
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.public Param<String> labelCol()
public String getLabelCol()
public Param<String> featuresCol()
public String getFeaturesCol()
public Param<String> predictionCol()
public String getPredictionCol()