Skip to main content

Module model_selection

Module model_selection 

Source
Expand description

Model selection: train_test_split, KFold, cross_val_score. Model-selection utilities mirroring sklearn.model_selection.

Provides train/test splitting, K-fold cross-validation splitters, and a generic cross_val_score that works with any Predictor

  • Clone estimator. A shared private PRNG lives in [rng].

Re-exports§

pub use cross_val::cross_val_score;
pub use kfold::KFold;
pub use kfold::StratifiedKFold;
pub use split::train_test_split;
pub use split::TrainTestSplit;

Modules§

cross_val
Cross-validation scoring driver. Cross-validation scoring driver mirroring sklearn.model_selection.cross_val_score.
kfold
K-fold and stratified K-fold splitters. K-fold and stratified K-fold cross-validation splitters.
rng
Shared deterministic PRNG (xorshift64). Deterministic seedable PRNG (xorshift64), shared by model_selection and decomposition::randomized_svd.
split
Train/test split. Train/test split utilities mirroring sklearn.model_selection.train_test_split.