DOC mention that train_test_split returns a random split