Hugging Face · Datasets

Load a dataset

Load the training split of a dataset from the Hub.

from datasets import load_dataset

ds = load_dataset("imdb", split="train")

Official Hugging Face documentation

More datasets commands

All datasets commands · Hugging Face cheatsheet

Was this page helpful?