Class: TrainValidationSplit

eclairjs/ml/tuning. TrainValidationSplit

Validation for hyper-parameter tuning. Randomly splits the input dataset into train and validation sets, and uses evaluation metric on the validation set to select the best model. Similar to CrossValidator, but only splits the set once.

Constructor

new TrainValidationSplit(uidopt)

Parameters:
Name Type Attributes Description
uid string <optional>
Source:

Extends

Methods

copy(extra) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
extra module:eclairjs/ml/param.ParamMap
Overrides:
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

extractParamMap() → {module:eclairjs/ml/param.ParamMap}

Inherited From:
Source:
Returns:
Type
module:eclairjs/ml/param.ParamMap

fit(dataset) → {module:eclairjs/ml/tuning.TrainValidationSplitModel}

Parameters:
Name Type Description
dataset module:eclairjs/sql.Dataset
Overrides:
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplitModel

load(path) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
path string
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

read() → {module:eclairjs/ml/util.MLReader}

Source:
Returns:
Type
module:eclairjs/ml/util.MLReader

setEstimator(value) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
value module:eclairjs/ml.Estimator
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

setEstimatorParamMaps(value) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
value Array.<ParamMap>
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

setEvaluator(value) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
value module:eclairjs/ml/evaluation.Evaluator
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

setSeed(value) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

setTrainRatio(value) → {module:eclairjs/ml/tuning.TrainValidationSplit}

Parameters:
Name Type Description
value number
Source:
Returns:
Type
module:eclairjs/ml/tuning.TrainValidationSplit

transformSchema(schema) → {module:eclairjs/sql/types.StructType}

Parameters:
Name Type Description
schema module:eclairjs/sql/types.StructType
Overrides:
Source:
Returns:
Type
module:eclairjs/sql/types.StructType

uid() → {string}

An immutable unique ID for the object and its derivatives.
Source:
Returns:
Type
string

write() → {module:eclairjs/ml/util.MLWriter}

Source:
Returns:
Type
module:eclairjs/ml/util.MLWriter