Skip to main content

train_test_split

Function train_test_split 

Source
pub fn train_test_split<T>(
    cont: &mut [T],
    train: f32,
    test: f32,
) -> Vec<&mut [T]>
where T: Clone,
Expand description

Generate train-test splits. Wrapper around split_parts

ยงExamples

use rand_split::train_test_split;

let mut cont = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13];
let total_len = cont.len();
let result = train_test_split(&mut cont, 0.8, 0.2);
assert_eq!(
    result.iter().map(|inner| inner.len()).sum::<usize>(),
    total_len
);