Struct RepeatedStratifiedKFold 

pub struct RepeatedStratifiedKFold<L> { /* private fields */ }

Repeated stratified K-fold.

The stratified analogue of RepeatedKFold: runs stratified K-fold n_repeats times with different seeds, preserving class proportions in every fold of every repeat. Total splits = n_repeats * n_splits.

use ndarray::Array1;
use model_selection_rs::splitters::{CvSplitter, RepeatedStratifiedKFold};

let y = Array1::from(vec![0, 1, 0, 1, 0, 1, 0, 1, 0, 1]);
let rskf = RepeatedStratifiedKFold::new(2, 3, 0, &y).unwrap();
assert_eq!(rskf.n_splits(), 6);

Implementations

impl<L: Eq + Hash + Clone> RepeatedStratifiedKFold<L>

pub fn new( n_splits: usize, n_repeats: usize, base_seed: u64, y: &Array1<L>, ) -> Result<Self>

Create a RepeatedStratifiedKFold over labels y.

Errors

Returns ModelSelectionError::InvalidSplitCount if n_splits < 2 or n_repeats < 1.

Trait Implementations

impl<L: Clone> Clone for RepeatedStratifiedKFold<L>

fn clone(&self) -> RepeatedStratifiedKFold<L>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<L: Eq + Hash + Clone> CvSplitter for RepeatedStratifiedKFold<L>

fn split(&self, n_samples: usize) -> Result<Vec<(Vec<usize>, Vec<usize>)>>

Produce every (train_indices, test_indices) pair for n_samples rows.

fn n_splits(&self) -> usize

Number of splits this strategy yields.

impl<L: Debug> Debug for RepeatedStratifiedKFold<L>

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

Formats the value using the given formatter.