Struct StratifiedGroupKFold 

pub struct StratifiedGroupKFold { /* private fields */ }

Stratified Group K-Fold cross-validator

This is a variation of GroupKFold that attempts to preserve the percentage of samples for each class while ensuring that the same group is not in both training and testing sets.

Implementations

impl StratifiedGroupKFold

pub fn new(n_splits: usize) -> Self

Create a new StratifiedGroupKFold cross-validator

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

Set whether to shuffle the groups before splitting

pub fn random_state(self, seed: u64) -> Self

Set the random state for shuffling

pub fn split_with_groups_and_labels( &self, n_samples: usize, y: &Array1<i32>, groups: &Array1<i32>, ) -> Vec<(Vec<usize>, Vec<usize>)>

Split data into train/test sets based on stratification and groups

Trait Implementations

impl Clone for StratifiedGroupKFold

fn clone(&self) -> StratifiedGroupKFold

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CrossValidator for StratifiedGroupKFold

fn n_splits(&self) -> usize

Returns the number of splits

fn split( &self, _n_samples: usize, _y: Option<&Array1<i32>>, ) -> Vec<(Vec<usize>, Vec<usize>)>

Generate train/test indices for cross-validation

impl Debug for StratifiedGroupKFold

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

Formats the value using the given formatter.