Struct KFold 

pub struct KFold {
    pub n_splits: usize,
    pub shuffle: bool,
    pub random_seed: u64,
}

K-fold cross-validation.

Splits the data into K equally-sized folds. Each fold is used once as validation while the K-1 remaining folds form the training set.

Fields

n_splits: usize

Number of folds.

shuffle: bool

Whether to shuffle the data before splitting.

random_seed: u64

Random seed for shuffling.

Implementations

impl KFold

pub fn new(n_splits: usize) -> TrainResult<Self>

Create a new K-fold splitter.

Arguments

• n_splits - Number of folds (must be >= 2)

pub fn with_shuffle(self, seed: u64) -> Self

Enable shuffling with a specific seed.

Trait Implementations

impl Clone for KFold

fn clone(&self) -> KFold

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl CrossValidationSplit for KFold

fn num_splits(&self) -> usize

Get the number of splits.

fn get_split( &self, fold: usize, n_samples: usize, ) -> TrainResult<(Vec<usize>, Vec<usize>)>

Get the train/validation indices for a specific fold.

impl Debug for KFold

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.