Struct KFold 

pub struct KFold { /* private fields */ }

Shuffled K-fold splitter with a fixed seed, so results are reproducible.

Implementations

impl KFold

pub fn new(n_folds: usize, seed: u64) -> Result<Self>

pub fn n_folds(&self) -> usize

pub fn split(&self, n: usize) -> Result<Vec<Fold>>

Split 0..n into n_folds disjoint test sets covering every index once.

pub fn split_stratified(&self, strata: &[usize]) -> Result<Vec<Fold>>

Stratified split: the rows of each stratum (e.g. treated / control) are shuffled and dealt round-robin over the folds, so every fold holds about the same share of each stratum. Fails if a stratum has fewer rows than there are folds.

Trait Implementations

impl Clone for KFold

fn clone(&self) -> Self

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.