Skip to main content

ttv_split

Function ttv_split 

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

Generate train-test-validation splits. Wrapper around split_parts

ยงExamples

use rand_split::ttv_split;

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