Struct StratifiedKFoldSpec 

pub struct StratifiedKFoldSpec {
    pub n_splits: usize,
    pub shuffle: bool,
    pub seed: Option<u64>,
}

Stratified K-fold: each sample is validated exactly once (OOF-safe like plain K-fold), but folds are balanced by a per-sample class label so every fold mirrors the overall class distribution. strata maps each sample id to its class label (identity-keyed metadata — never feature values).

Fields

n_splits: usize

shuffle: bool

seed: Option<u64>

Implementations

impl StratifiedKFoldSpec

pub fn split( &self, id: impl Into<String>, samples: &[SampleId], strata: &BTreeMap<SampleId, String>, ) -> Result<FoldSet, DagMlError>

Trait Implementations

impl Clone for StratifiedKFoldSpec

fn clone(&self) -> StratifiedKFoldSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StratifiedKFoldSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for StratifiedKFoldSpec

fn deserialize<__D>( __deserializer: __D, ) -> Result<StratifiedKFoldSpec, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for StratifiedKFoldSpec

impl PartialEq for StratifiedKFoldSpec

fn eq(&self, other: &StratifiedKFoldSpec) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for StratifiedKFoldSpec

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for StratifiedKFoldSpec