Struct DatasetSplit 

pub struct DatasetSplit {
    pub train: ArrowDataset,
    pub test: ArrowDataset,
    pub validation: Option<ArrowDataset>,
}

Dataset split with optional validation set

Fields

train: ArrowDataset

Training dataset (required)

test: ArrowDataset

Test/holdout dataset (required)

validation: Option<ArrowDataset>

Validation dataset (optional)

Implementations

impl DatasetSplit

pub fn new(train: ArrowDataset, test: ArrowDataset) -> Self

Create train/test split (no validation)

pub fn with_validation( train: ArrowDataset, test: ArrowDataset, validation: ArrowDataset, ) -> Self

Create train/test/validation split

pub fn train(&self) -> &ArrowDataset

Get training data

pub fn test(&self) -> &ArrowDataset

Get test data

pub fn validation(&self) -> Option<&ArrowDataset>

Get validation data (if present)

pub fn from_ratios( dataset: &ArrowDataset, train_ratio: f64, test_ratio: f64, val_ratio: Option<f64>, seed: Option<u64>, ) -> Result<Self>

Split dataset by ratios

Arguments

• dataset - Source dataset to split
• train_ratio - Fraction for training (0.0 to 1.0)
• test_ratio - Fraction for testing (0.0 to 1.0)
• val_ratio - Optional fraction for validation
• seed - Optional random seed for shuffling

Errors

Returns error if ratios don’t sum to 1.0 or dataset is empty

pub fn stratified( dataset: &ArrowDataset, label_column: &str, train_ratio: f64, test_ratio: f64, val_ratio: Option<f64>, seed: Option<u64>, ) -> Result<Self>

Stratified split preserving label distribution

Arguments

• dataset - Source dataset to split
• label_column - Name of the label/target column
• train_ratio - Fraction for training
• test_ratio - Fraction for testing
• val_ratio - Optional fraction for validation
• seed - Optional random seed

Errors

Returns error if label column not found or ratios invalid

Trait Implementations

impl Clone for DatasetSplit

fn clone(&self) -> DatasetSplit

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DatasetSplit

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

Formats the value using the given formatter.