When prompted, enter your token to login: | |
from huggingface_hub import notebook_login | |
notebook_login() | |
Load IMDb dataset | |
Start by loading the IMDb dataset from the 🤗 Datasets library: | |
from datasets import load_dataset | |
imdb = load_dataset("imdb") | |
Then take a look at an example: | |
imdb["test"][0] | |
{ | |
"label": 0, | |
"text": "I love sci-fi and am willing to put up with a lot. |