当前位置：首页 » AI答疑

如何使用Rankify中的预检索数据集？

2025-08-28

1.2 K

准备工作

Rankify提供了40个预检索数据集，这些数据集可以从Hugging Face平台下载使用。

from rankify.dataset.dataset import Dataset

查看所有预置数据集：

Dataset.available_dataset()

以BM25检索器和nq-dev数据集为例：

dataset = Dataset(retriever="bm25", dataset_name="nq-dev", n_docs=100)
documents = dataset.download(force_download=False)

若已有下载的数据集文件：

documents = Dataset.load_dataset('./bm25_nq_dev.json', 100)