Struct KFold 

pub struct KFold {
    pub n_splits: usize,
    pub shuffle: bool,
    pub seed: Option<u64>,
}

K-fold cross-validation splitter.

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

Example

use scirs2_core::data_split::KFold;

let kf = KFold::new(5, true, Some(42)).expect("kfold");
let splits: Vec<_> = kf.split(100).collect();
assert_eq!(splits.len(), 5);
for (train, test) in &splits {
    assert_eq!(train.len() + test.len(), 100);
}

Fields

n_splits: usize

Number of folds

shuffle: bool

Whether to shuffle before splitting

seed: Option<u64>

Random seed

Implementations

impl KFold

pub fn new( n_splits: usize, shuffle: bool, seed: Option<u64>, ) -> CoreResult<Self>

Create a new KFold splitter.

pub fn split(&self, n_samples: usize) -> impl Iterator<Item = SplitIndices>

Generate splits for n_samples data points.

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 Debug for KFold

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

Formats the value using the given formatter.