use matten::Tensor;
use matten_mlprep::train_test_split;
fn main() {
let x = Tensor::new((0..10).map(|v| v as f64).collect(), &[5, 2]);
let (train, test) = train_test_split(&x, 0.6).expect("valid split"); println!("train {:?}: {:?}", train.shape(), train.as_slice());
println!("test {:?}: {:?}", test.shape(), test.as_slice());
}