public class TrainValidationSplit extends Estimator<TrainValidationSplitModel> implements Logging
CrossValidator
, but only splits the set once.Constructor and Description |
---|
TrainValidationSplit() |
TrainValidationSplit(String uid) |
Modifier and Type | Method and Description |
---|---|
TrainValidationSplit |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
Param<Estimator<?>> |
estimator()
param for the estimator to be validated
|
Param<ParamMap[]> |
estimatorParamMaps()
param for estimator param maps
|
Param<Evaluator> |
evaluator()
param for the evaluator used to select hyper-parameters that maximize the validated metric
|
TrainValidationSplitModel |
fit(DataFrame dataset)
Fits a model to the input data.
|
Estimator<?> |
getEstimator() |
ParamMap[] |
getEstimatorParamMaps() |
Evaluator |
getEvaluator() |
double |
getTrainRatio() |
TrainValidationSplit |
setEstimator(Estimator<?> value) |
TrainValidationSplit |
setEstimatorParamMaps(ParamMap[] value) |
TrainValidationSplit |
setEvaluator(Evaluator value) |
TrainValidationSplit |
setTrainRatio(double value) |
DoubleParam |
trainRatio()
Param for ratio between train and validation data.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
void |
validateParams() |
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
toString
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public TrainValidationSplit(String uid)
public TrainValidationSplit()
public String uid()
Identifiable
uid
in interface Identifiable
public TrainValidationSplit setEstimator(Estimator<?> value)
public TrainValidationSplit setEstimatorParamMaps(ParamMap[] value)
public TrainValidationSplit setEvaluator(Evaluator value)
public TrainValidationSplit setTrainRatio(double value)
public TrainValidationSplitModel fit(DataFrame dataset)
Estimator
fit
in class Estimator<TrainValidationSplitModel>
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()
validateParams
in interface Params
public TrainValidationSplit copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Estimator<TrainValidationSplitModel>
extra
- (undocumented)defaultCopy()
public DoubleParam trainRatio()
public double getTrainRatio()
public Param<Estimator<?>> estimator()
public Estimator<?> getEstimator()
public Param<ParamMap[]> estimatorParamMaps()
public ParamMap[] getEstimatorParamMaps()
public Param<Evaluator> evaluator()
public Evaluator getEvaluator()