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 |
---|---|
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.
|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copyValues, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams
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 Learner copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Estimator<M extends PredictionModel<FeaturesType,M>>
extra
- (undocumented)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.