train_test_split splits data stratify samlping code example Example: train_test_split example train_test_split example