tuning

Type Members

1. class CrossValidator extends Estimator[CrossValidatorModel] with CrossValidatorParams with HasParallelism with HasCollectSubModels with MLWritable with Logging

   

   K-fold cross validation performs model selection by splitting the dataset into a set of non-overlapping randomly partitioned folds which are used as separate training and test datasets e.g., with k=3 folds, K-fold cross validation will generate 3 (training, test) dataset pairs, each of which uses 2/3 of the data for training and 1/3 for testing.

   K-fold cross validation performs model selection by splitting the dataset into a set of non-overlapping randomly partitioned folds which are used as separate training and test datasets e.g., with k=3 folds, K-fold cross validation will generate 3 (training, test) dataset pairs, each of which uses 2/3 of the data for training and 1/3 for testing. Each fold is used as the test set exactly once.

   Annotations

   @Since( "1.2.0" )

2. class CrossValidatorModel extends Model[CrossValidatorModel] with CrossValidatorParams with MLWritable

   

   CrossValidatorModel contains the model with the highest average cross-validation metric across folds and uses this model to transform input data.

   CrossValidatorModel contains the model with the highest average cross-validation metric across folds and uses this model to transform input data. CrossValidatorModel also tracks the metrics for each param map evaluated.

   Annotations

   @Since( "1.2.0" )

3. class ParamGridBuilder extends AnyRef

   

   Builder for a param grid used in grid search-based model selection.

   Builder for a param grid used in grid search-based model selection.

   Annotations

   @Since( "1.2.0" )

4. class TrainValidationSplit extends Estimator[TrainValidationSplitModel] with TrainValidationSplitParams with HasParallelism with HasCollectSubModels with MLWritable with Logging

   

   Validation for hyper-parameter tuning.

   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.

   Annotations

   @Since( "1.5.0" )

5. class TrainValidationSplitModel extends Model[TrainValidationSplitModel] with TrainValidationSplitParams with MLWritable

   

   Model from train validation split.

   Model from train validation split.

   Annotations

   @Since( "1.5.0" )