I have a dataset with multiple labels, ie for each X I have 2 y and I need to split into train and test set.
I tried with the sklearn function train_test_split():
