Struct KFold 

pub struct KFold { /* private fields */ }

K-Fold cross-validation split indices generator

Implementations

impl KFold

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

Create a new K-Fold splitter

pub fn n_splits(&self) -> usize

Get the number of splits

pub fn shuffle(&self) -> bool

Get whether shuffling is enabled

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 KFold

fn clone(&self) -> KFold

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.