[][src]Function rand_split::train_test_split

pub fn train_test_split<T>(
    cont: &[T],
    train: f32,
    test: f32
) -> Result<Vec<Vec<T>>, &'static str> where
    T: Clone

Generate train-test splits. Wrapper around split_parts

Examples

use rand_split::train_test_split;

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