pub fn train_test_split( features: &Array2<f64>, labels: &Array1<usize>, test_ratio: f64, shuffle: bool, ) -> Result<(Array2<f64>, Array1<usize>, Array2<f64>, Array1<usize>)>
Split data into train and test sets