Function train_test_split 

pub fn train_test_split<D>(
    dataset: D,
    ratios: &[f64],
    seed: u64,
) -> Vec<SubsetDataset<D>>
where
    D: Dataset + Clone,

Split a dataset into (train, val) or (train, val, test) subsets.

Returns SubsetDataset views over the original dataset.

Arguments

• dataset — the source dataset
• ratios — slice of 2 or 3 floats that sum to 1.0, e.g. [0.8, 0.2] or [0.7, 0.15, 0.15]
• seed — random seed for reproducible shuffling of indices