Struct StratifiedKFold 

pub struct StratifiedKFold {
    pub n_splits: usize,
    pub shuffle: bool,
    pub random_state: Option<u64>,
}

Stratified K-Fold cross-validation splitter

Fields

n_splits: usize

Number of folds

shuffle: bool

Whether to shuffle data before splitting

random_state: Option<u64>

Random seed for shuffling

Implementations

impl StratifiedKFold

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

Create a new Stratified K-Fold splitter

pub fn split( &self, y: &Array1<i32>, ) -> Result<Vec<(Vec<usize>, Vec<usize>)>, SklearsError>

Generate stratified train/test splits

Trait Implementations

impl Clone for StratifiedKFold

fn clone(&self) -> StratifiedKFold

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StratifiedKFold

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

Formats the value using the given formatter.