Error with sklearn train_test_split
{'login': 'ClonedOne', 'id': 6853743, 'node_id': 'MDQ6VXNlcjY4NTM3NDM=', 'avatar_url': 'https://avatars.githubusercontent.com/u/6853743?v=4', 'gravatar_id': '', 'url': 'https://api.github.com/users/ClonedOne', 'html_url': 'https://github.com/ClonedOne', 'followers_url': 'https://api.github.com/users/ClonedOne/followers', 'following_url': 'https://api.github.com/users/ClonedOne/following{/other_user}', 'gists_url': 'https://api.github.com/users/ClonedOne/gists{/gist_id}', 'starred_url': 'https://api.github.com/users/ClonedOne/starred{/owner}{/repo}', 'subscriptions_url': 'https://api.github.com/users/ClonedOne/subscriptions', 'organizations_url': 'https://api.github.com/users/ClonedOne/orgs', 'repos_url': 'https://api.github.com/users/ClonedOne/repos', 'events_url': 'https://api.github.com/users/ClonedOne/events{/privacy}', 'received_events_url': 'https://api.github.com/users/ClonedOne/received_events', 'type': 'User', 'site_admin': False}
[{'id': 1935892871, 'node_id': 'MDU6TGFiZWwxOTM1ODkyODcx', 'url': 'https://api.github.com/repos/huggingface/datasets/labels/enhancement', 'name': 'enhancement', 'color': 'a2eeef', 'default': True, 'description': 'New feature or request'}]
[]
Indeed. Probably we will want to have a similar method directly in the library Related: #166 
It would be nice if we could use sklearn `train_test_split` to quickly generate subsets from the dataset objects returned by `nlp.load_dataset`. At the moment the code:

```python
data = nlp.load_dataset('imdb', cache_dir=data_cache)
f_half, s_half = train_test_split(data['train'], test_size=0.5, random_state=seed)
```
throws:
```
ValueError: Can only get row(s) (int or slice) or columns (string).
```
It's not a big deal, since there are other ways to split the data, but it would be a cool thing to have.
{'url': 'https://api.github.com/repos/huggingface/datasets/issues/147/reactions', 'total_count': 0, '+1': 0, '-1': 0, 'laugh': 0, 'hooray': 0, 'confused': 0, 'heart': 0, 'rocket': 0, 'eyes': 0}