【免费下载链接】super_glue
🧩 想在 3 分钟内用上SuperGLUE基准测试数据吗?本文面向新手,用Hugging Face datasets库,仅 3 步就能加载SuperGLUE Parquet数据并查看内容,还附上 11 个子任务的清单和常见报错避坑指南。
SuperGLUE 是 2019 年发布的"更困难版"GLUE 语言理解基准,涵盖自然语言推理、阅读理解、词义消歧、共指消解等 11 个任务。Hugging Face 官方仓库中,每个任务的数据都按 Parquet 格式分片存放,datasets 库会自动下载并转换,无需手动处理。
准备工作:安装 datasets 库
只需要一行命令:
pip install -U datasets💡 如果你用的是 Conda 环境,可以把
pip换成conda,或改用pip install避免依赖冲突。
确认安装成功:
python -c "import datasets; print(datasets.__version__)"能打印出版本号(如3.x.x)即可开始下一步。
第 1 步:一行代码加载 SuperGLUE 数据
SuperGLUE 在 Hugging Face Hub 上由config_name区分 11 个子任务,所以加载时要传两个参数:数据集名 + 配置名。
from datasets import load_dataset dataset = load_dataset("super_glue", "boolq") print(dataset)运行后会看到类似输出:
DatasetDict({ train: Dataset({ features: ['question', 'passage', 'idx', 'label'], num_rows: 9427 }), validation: Dataset({...}), test: Dataset({...}) })几个关键参数说明:
| 参数 | 含义 | 示例 |
|---|---|---|
| 第 1 个参数 | 数据集名称 | "super_glue" |
| 第 2 个参数(config) | 子任务配置名 | "boolq"、"rte"、"copa" |
split | 只加载某个数据划分 | split="validation" |
streaming | 流式读取,省内存 | streaming=True |
只想加载验证集?加一个split参数即可:
dataset = load_dataset("super_glue", "cb", split="validation")第 2 步:查看字段结构与数据样例
数据集加载后自带features属性,能列出所有字段名和类型。以boolq为例,它的字段是question、passage、idx、label(详见 README.md 的 Data Fields 部分)。
print(dataset["train"].features) print(dataset["train"][0])label默认显示为整数,用int2str可以还原成语义标签:
d = dataset["train"][0] print(d["question"]) print(dataset["train"].features["label"].int2str(d["label"])) # True / False不同子任务的字段略有差异,例如:
- cb / axg / rte:
premise(前提)、hypothesis(假设)、label - copa:
premise、choice1、choice2、question - record:
passage、query、entities、answers - wsc / wsc.fixed:
text、span1_text、span2_text、span1_index、span2_index
⚠️注意:
test划分只有axb和axg例外——它们只提供了 test 集。而cb、copa、record、rte等任务的 test 集在 Hugging Face 上通常不含 label 字段(用于官方排行榜评测),加载 test 集时请先确认字段是否存在。
第 3 步:转换格式,直接喂给模型
训练微调前,通常把数据转成 PyTorch 格式:
dataset.set_format("torch")如果数据集太大(比如record有 10 万+ 训练样本、约 200 MB),推荐流式读取,不占本地磁盘:
dataset = load_dataset("super_glue", "record", split="train", streaming=True) dataset = dataset.shuffle(seed=42).take(100) dataset.set_format("torch")streaming=True配合take()、shuffle(),是处理大 Parquet 数据集最快的配置方法之一。
SuperGLUE 的 11 个子任务一览 📊
Hugging Face 仓库为每个任务单独建目录存放 Parquet 文件(文件名格式为split-00000-of-00001.parquet),行数如下表所示:
| 任务 | 说明 | 训练集 | 验证集 | 测试集 |
|---|---|---|---|---|
| axb | 自然语言推理(AX-B) | — | — | 1,104 |
| axg | 自然语言推理(AX-G) | — | — | 356 |
| boolq | 阅读理解,是非判断 | 9,427 | 3,270 | 3,245 |
| cb | 承诺关系推断 | 250 | 56 | 250 |
| copa | 常识因果推理 | 400 | 100 | 500 |
| multirc | 多句阅读理解 | 27,243 | 4,848 | 9,693 |
| record | 常识阅读理解 | 100,730 | 10,000 | 10,000 |
| rte | 文本蕴含识别 | 2,490 | 277 | 3,000 |
| wic | 词义上下文消歧 | 5,428 | 638 | 1,400 |
| wsc | Winograd 共指消解 | 554 | 104 | 146 |
| wsc.fixed | WSC 修正版 | 554 | 104 | 146 |
每个任务的 Parquet 文件都在对应子目录中,例如:
- boolq/train-00000-of-00001.parquet
- record/test-00000-of-00001.parquet
- wsc.fixed/validation-00000-of-00001.parquet
完整的字段类型、标签取值(如entailment、contradiction)和分片信息,都可以查阅 README.md 中的 dataset_info 元数据。
常见坑与快速解决方法 🛠️
报
BuilderConfig 'xxx' not found配置名写错了。SuperGLUE 只有上表 11 个合法配置名,注意wsc.fixed中的点号,不要写成wsc_fixed。test 集没有 label 列属于正常现象(官方评测集),调试时请用
train或validation划分。加载
record时内存告警改用streaming=True,或加载后dataset.select(range(2000))先取子集。想直接读本地 Parquet 文件不经过 Hugging Face 下载,可以用 PyArrow 直接打开:
import pyarrow.parquet as pq table = pq.read_table("boolq/train-00000-of-00001.parquet")clone 下来的仓库里 parquet 文件很小?这类镜像仓库的 Parquet 文件可能是 Git LFS 占位符,直接访问会报
Parquet magic bytes not found。请通过load_dataset在线加载,或执行git lfs pull拉取真实文件。
小结:3 步回顾 ✅
pip install -U datasets安装依赖;load_dataset("super_glue", "boolq")加载指定子任务;- 查看
features、取[0]看样例,set_format("torch")后进入训练流程。
SuperGLUE 数据规模从几百条(cb)到十几万条(record)不等,配合 datasets 库的流式读取和分片机制,新手也能轻松完成从"加载 Parquet 数据"到"模型微调"的第一步。
📚 更多资料:完整的任务定义、许可证与引用信息见 README.md,各任务的 Parquet 文件位于各任务同名目录下。
【免费下载链接】super_glue
相关推荐
终极指南:Hugging Face数据集查看器完整教程与快速上手
终极指南:Hugging Face数据集查看器完整教程与快速上手 Hugging Face数据集查看器是一个功能强大的开源工具,专门为数据科学家和机器学习工程师
大数据处理技术全攻略:系统设计面试题精选大数据篇
大数据处理技术全攻略:系统设计面试题精选大数据篇 系统设计面试中,大数据处理技术是考察的重点领域之一。本文将围绕GitHub加速计划中的system desig
离线语音合成新境界:ChatTTS-ui无网络部署实战指南
离线语音合成新境界:ChatTTS ui无网络部署实战指南 在网络安全日益重要的今天,离线语音合成系统成为许多场景的刚需。想象一下:在涉密实验室里,你需要将技术
AI 技能人工智能AI 评测模型评测
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考