public class CrossValidator extends Estimator<CrossValidatorModel> implements Logging
Constructor and Description |
---|
CrossValidator() |
CrossValidator(String uid) |
Modifier and Type | Method and Description |
---|---|
Param<Estimator<?>> |
estimator()
param for the estimator to be cross-validated
|
Param<ParamMap[]> |
estimatorParamMaps()
param for estimator param maps
|
Param<Evaluator> |
evaluator()
param for the evaluator used to select hyper-parameters that maximize the cross-validated
metric
|
CrossValidatorModel |
fit(DataFrame dataset)
Fits a model to the input data.
|
Estimator<?> |
getEstimator() |
ParamMap[] |
getEstimatorParamMaps() |
Evaluator |
getEvaluator() |
int |
getNumFolds() |
IntParam |
numFolds()
Param for number of folds for cross validation.
|
CrossValidator |
setEstimator(Estimator<?> value) |
CrossValidator |
setEstimatorParamMaps(ParamMap[] value) |
CrossValidator |
setEvaluator(Evaluator value) |
CrossValidator |
setNumFolds(int value) |
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid() |
void |
validateParams()
Validates parameter values stored internally.
|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copy, copyValues, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public CrossValidator(String uid)
public CrossValidator()
public String uid()
public CrossValidator setEstimator(Estimator<?> value)
public CrossValidator setEstimatorParamMaps(ParamMap[] value)
public CrossValidator setEvaluator(Evaluator value)
public CrossValidator setNumFolds(int value)
public CrossValidatorModel fit(DataFrame dataset)
Estimator
fit
in class Estimator<CrossValidatorModel>
dataset
- (undocumented)public StructType transformSchema(StructType schema)
PipelineStage
Derives the output schema from the input schema.
transformSchema
in class PipelineStage
schema
- (undocumented)public void validateParams()
Params
This only needs to check for interactions between parameters.
Parameter value checks which do not depend on other parameters are handled by
Param.validate()
. This method does not handle input/output column parameters;
those are checked during schema validation.
validateParams
in interface Params
public Param<Estimator<?>> estimator()
public Estimator<?> getEstimator()
public Param<ParamMap[]> estimatorParamMaps()
public ParamMap[] getEstimatorParamMaps()
public Param<Evaluator> evaluator()
public Evaluator getEvaluator()
public IntParam numFolds()
public int getNumFolds()