Hub documentation

使用 🤗 Datasets

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

使用 🤗 Datasets

在 Hugging Face Hub 上找到感兴趣的 dataset 后,你可以使用 🤗 Datasets 来加载该 dataset。点击 Use this dataset 按钮 即可复制加载 dataset 的代码。

首先,您需要使用 Hugging Face 账户登录,例如使用以下命令:

hf auth login

然后,您可以使用

from datasets import load_dataset

dataset = load_dataset("username/my_dataset")

# or load the separate splits if the dataset has train/validation/test splits
train_dataset = load_dataset("username/my_dataset", split="train")
valid_dataset = load_dataset("username/my_dataset", split="validation")
test_dataset  = load_dataset("username/my_dataset", split="test")

您还可以将数据集上传到 Hugging Face Hub:

my_new_dataset.push_to_hub("username/my_new_dataset")

这将创建一个数据集仓库 username/my_new_dataset,其中包含以 Parquet 格式存储的 Dataset,您可以稍后重新加载。

如需了解有关使用 🤗 Datasets 的更多信息,请查阅 🤗 Datasets 文档中提供的 教程 和 操作指南。

Update on GitHub