public abstract class Estimator<M extends Model<M>> extends PipelineStage implements Params
Constructor and Description |
---|
Estimator() |
Modifier and Type | Method and Description |
---|---|
abstract M |
fit(DataFrame dataset,
ParamMap paramMap)
Fits a single model to the input data with provided parameter map.
|
scala.collection.Seq<M> |
fit(DataFrame dataset,
ParamMap[] paramMaps)
Fits multiple models to the input data with multiple sets of parameters.
|
M |
fit(DataFrame dataset,
ParamPair<?>... paramPairs)
Fits a single model to the input data with optional parameters.
|
M |
fit(DataFrame dataset,
scala.collection.Seq<ParamPair<?>> paramPairs)
Fits a single model to the input data with optional parameters.
|
transformSchema
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
addOutputColumn, checkInputColumn, explainParams, get, getParam, isSet, paramMap, params, set, set, validate, validate
uid
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public M fit(DataFrame dataset, ParamPair<?>... paramPairs)
dataset
- input datasetparamPairs
- Optional list of param pairs.
These values override any specified in this Estimator's embedded ParamMap.public M fit(DataFrame dataset, scala.collection.Seq<ParamPair<?>> paramPairs)
dataset
- input datasetparamPairs
- Optional list of param pairs.
These values override any specified in this Estimator's embedded ParamMap.public abstract M fit(DataFrame dataset, ParamMap paramMap)
dataset
- input datasetparamMap
- Parameter map.
These values override any specified in this Estimator's embedded ParamMap.public scala.collection.Seq<M> fit(DataFrame dataset, ParamMap[] paramMaps)
dataset
- input datasetparamMaps
- An array of parameter maps.
These values override any specified in this Estimator's embedded ParamMap.