pub fn train_test_split<D: Dimension + RemoveAxis, D2: Dimension + RemoveAxis, Feature: Clone, Label: Clone>(
arr: &Array<Feature, D>,
y: &Array<Label, D2>,
split: f64,
) -> (Array<Feature, D>, Array<Feature, D>, Array<Label, D2>, Array<Label, D2>)