Struct smartcore::model_selection::KFold

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

K-Folds cross-validator

Fields

n_splits: usize

Number of folds. Must be at least 2.

shuffle: bool

Whether to shuffle the data before splitting into batches

Implementations

impl KFold

pub fn with_n_splits(self, n_splits: usize) -> Self

Number of folds. Must be at least 2.

pub fn with_shuffle(self, shuffle: bool) -> Self

Whether to shuffle the data before splitting into batches

Trait Implementations

impl BaseKFold for KFold

Abstract class for all KFold functionalities

type Output = KFoldIter

An iterator over indices that split data into training and test set.

pub fn n_splits(&self) -> usize

Returns the number of splits

pub fn split<T: RealNumber, M: Matrix<T>>(&self, x: &M) -> Self::Output

Return a tuple containing the the training set indices for that split and the testing set indices for that split.

impl Default for KFold

pub fn default() -> KFold

Returns the "default value" for a type.