Struct StratifiedKFold 

pub struct StratifiedKFold { /* private fields */ }

Stratified K-Fold cross-validation split indices generator Ensures each fold has approximately the same percentage of samples of each class

Implementations

impl StratifiedKFold

pub fn new( labels: &Array1<usize>, n_splits: usize, shuffle: bool, ) -> Result<Self>

Create a new Stratified K-Fold splitter

pub fn n_splits(&self) -> usize

Get the number of splits

pub fn get_fold(&self, fold: usize) -> Result<(Vec<usize>, Vec<usize>)>

Get train and test indices for a specific fold

pub fn split( &self, features: &Array2<f64>, labels: &Array1<usize>, fold: usize, ) -> Result<(Array2<f64>, Array1<usize>, Array2<f64>, Array1<usize>)>

Split features and labels for a specific fold

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.