Hyukkyu/beir-msmarco

---
license: apache-2.0
task_categories:
- information-retrieval
- text-retrieval
tags:
- beir
- msmarco
- information-retrieval
- retrieval
- search
dataset_info:
- config_name: corpus
features:
- name: _id
dtype: string
- name: title
dtype: string
- name: metadata
dtype: 'null'
- name: text
dtype: string
splits:
- name: train
num_bytes: 3149969815
num_examples: 8841823
download_size: 1641359367
dataset_size: 3149969815
- config_name: queries
features:
- name: _id
dtype: string
- name: metadata
dtype: 'null'
- name: text
dtype: string
splits:
- name: train
num_bytes: 24100662
num_examples: 509962
download_size: 15448970
dataset_size: 24100662
configs:
- config_name: corpus
data_files:
- split: train
path: corpus/train-*
- config_name: queries
data_files:
- split: train
path: queries/train-*
---

# BEIR MSMARCO Dataset (Migrated)

This is a migrated version of BeIR/msmarco that is compatible with datasets library 4.0.0+.

## Dataset Description

This dataset contains the msmarco dataset from the BEIR benchmark, converted from the old script-based format to Parquet format.

## Dataset Structure

### Queries

- **Split 'queries'**: 509,962 examples
- Features: ['_id', 'text', 'metadata']

- **Total examples**: 509,962

### Corpus

- **Split 'corpus'**: 8,841,823 examples
- Features: ['_id', 'title', 'text', 'metadata']

- **Total examples**: 8,841,823

## Usage

```python
from datasets import load_dataset

# Load queries (split: queries)
queries = load_dataset("Hyukkyu/beir-msmarco", "queries", split="queries")

# Load corpus (split: corpus)
corpus = load_dataset("Hyukkyu/beir-msmarco", "corpus", split="corpus")
```

## Available Splits

### Queries
- `queries`: 509,962 examples

### Corpus
- `corpus`: 8,841,823 examples

## Original Dataset

This dataset is migrated from: BeIR/msmarco

## Citation

If you use this dataset, please cite the original BEIR paper:

```bibtex
@article{thakur2021beir,
title={BEIR: A Heterogeneous Benchmark for Zero-shot Evaluation of Information Retrieval Models},
author={Thakur, Nandan and Reimers, Nils and Ruckle, Andreas and Srivastava, Abhishek and Gurevych, Iryna},
journal={arXiv preprint arXiv:2104.08663},
year={2021}
}
```