beir

Stay organized with collections Save and categorize content based on your preferences.

  • Description:

BEIR is a heterogeneous benchmark containing diverse IR tasks. It also provides a common and easy framework for evaluation of your NLP-based retrieval models within the benchmark.

FeaturesDict({
    'passage': Text(shape=(), dtype=tf.string),
    'passage_id': Text(shape=(), dtype=tf.string),
    'passage_metadata': Text(shape=(), dtype=tf.string),
    'query': Text(shape=(), dtype=tf.string),
    'query_id': Text(shape=(), dtype=tf.string),
    'query_metadata': Text(shape=(), dtype=tf.string),
    'score': tf.float32,
})
  • Feature documentation:
Feature Class Shape Dtype Description
FeaturesDict
passage Text tf.string
passage_id Text tf.string
passage_metadata Text tf.string
query Text tf.string
query_id Text tf.string
query_metadata Text tf.string
score Tensor tf.float32
@inproceedings{
    thakur2021beir,
    title={ {BEIR}: A Heterogeneous Benchmark for Zero-shot Evaluation of Information Retrieval Models},
    author={Nandan Thakur and Nils Reimers and Andreas R{"u}ckl{'e} and Abhishek Srivastava and Iryna Gurevych},
    booktitle={Thirty-fifth Conference on Neural Information Processing Systems Datasets and Benchmarks Track (Round 2)},
    year={2021},
    url={https://openreview.net/forum?id=wCu6T5xFjeJ}
}

beir/msmarco (default config)

  • Download size: 1.01 GiB

  • Dataset size: 4.53 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 8,841,823
'query' 509,962
'test' 9,139
'train' 516,472
'validation' 7,433

beir/trec_covid

  • Download size: 70.45 MiB

  • Dataset size: 292.04 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 171,332
'query' 50
'test' 35,480

beir/nfcorpus

  • Download size: 2.34 MiB

  • Dataset size: 24.94 MiB

  • Auto-cached (documentation): Yes

  • Splits:

Split Examples
'passage' 3,633
'query' 3,237
'test' 3,128
'train' 3,588
'validation' 3,119

beir/nq

  • Download size: 475.22 MiB

  • Dataset size: 1.66 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 2,681,468
'query' 3,452
'test' 4,201

beir/hotpotqa

  • Download size: 623.73 MiB

  • Dataset size: 2.64 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 5,233,329
'query' 97,852
'test' 13,783
'train' 101,307
'validation' 10,335

beir/fiqa

  • Download size: 17.12 MiB

  • Dataset size: 73.39 MiB

  • Auto-cached (documentation): Yes

  • Splits:

Split Examples
'passage' 57,638
'query' 6,648
'test' 1,706
'train' 14,166
'validation' 1,238

beir/arguana

  • Download size: 3.60 MiB

  • Dataset size: 15.08 MiB

  • Auto-cached (documentation): Yes

  • Splits:

Split Examples
'passage' 8,674
'query' 1,406
'test' 1,401

beir/webis_touche2020

  • Download size: 216.61 MiB

  • Dataset size: 747.57 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 382,545
'query' 49
'test' 2,099

beir/cqadupstack.android

  • Download size: 4.98 GiB

  • Dataset size: 883.49 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 22,998
'query' 699
'test' 1,696

beir/cqadupstack.english

  • Download size: 4.98 GiB

  • Dataset size: 3.78 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 40,221
'query' 1,570
'test' 3,765

beir/cqadupstack.gaming

  • Download size: 4.98 GiB

  • Dataset size: 2.64 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 45,301
'query' 1,595
'test' 2,263

beir/cqadupstack.gis

  • Download size: 4.98 GiB

  • Dataset size: 1.47 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 37,637
'query' 885
'test' 1,114

beir/cqadupstack.mathematica

  • Download size: 4.98 GiB

  • Dataset size: 1.58 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 16,705
'query' 804
'test' 1,358

beir/cqadupstack.physics

  • Download size: 4.98 GiB

  • Dataset size: 2.02 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 38,316
'query' 1,039
'test' 1,933

beir/cqadupstack.programmers

  • Download size: 4.98 GiB

  • Dataset size: 2.14 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 32,176
'query' 876
'test' 1,675

beir/cqadupstack.stats

  • Download size: 4.98 GiB

  • Dataset size: 941.07 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 42,269
'query' 652
'test' 913

beir/cqadupstack.tex

  • Download size: 4.98 GiB

  • Dataset size: 21.75 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 68,184
'query' 2,906
'test' 5,154

beir/cqadupstack.unix

  • Download size: 4.98 GiB

  • Dataset size: 2.04 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 47,382
'query' 1,072
'test' 1,693

beir/cqadupstack.webmasters

  • Download size: 4.98 GiB

  • Dataset size: 650.08 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 17,405
'query' 506
'test' 1,395

beir/cqadupstack.wordpress

  • Download size: 4.98 GiB

  • Dataset size: 726.15 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 48,605
'query' 541
'test' 744

beir/quora

  • Download size: 15.12 MiB

  • Dataset size: 121.33 MiB

  • Auto-cached (documentation): Yes

  • Splits:

Split Examples
'passage' 522,931
'query' 15,000
'test' 15,675
'validation' 7,626

beir/dbpedia_entity

  • Download size: 609.67 MiB

  • Dataset size: 2.47 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 4,635,922
'query' 467
'test' 40,724
'validation' 5,658

beir/scidocs

  • Download size: 135.87 MiB

  • Dataset size: 573.04 MiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 25,657
'query' 1,000
'test' 25,657

beir/fever

  • Download size: 1.15 GiB

  • Dataset size: 3.70 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 5,416,568
'query' 123,142
'test' 1,499
'train' 12,547
'validation' 1,460

beir/climate_fever

  • Download size: 1.14 GiB

  • Dataset size: 3.64 GiB

  • Auto-cached (documentation): No

  • Splits:

Split Examples
'passage' 5,416,593
'query' 1,535
'test' 1,344

beir/scifact

  • Download size: 2.69 MiB

  • Dataset size: 10.16 MiB

  • Auto-cached (documentation): Yes

  • Splits:

Split Examples
'passage' 5,183
'query' 1,109
'test' 283
'train' 565