beir

  • 説明:

BEIR は、さまざまな IR タスクを含む異種ベンチマークです。また、ベンチマーク内で NLP ベースの検索モデルを評価するための共通の簡単なフレームワークも提供します。

FeaturesDict({
    'passage': Text(shape=(), dtype=string),
    'passage_id': Text(shape=(), dtype=string),
    'passage_metadata': Text(shape=(), dtype=string),
    'query': Text(shape=(), dtype=string),
    'query_id': Text(shape=(), dtype=string),
    'query_metadata': Text(shape=(), dtype=string),
    'score': float32,
})
  • 機能のドキュメント:
特徴クラスDtype説明
特徴辞書
通路文章ストリング
パッセージ_id文章ストリング
パッセージ_メタデータ文章ストリング
クエリ文章ストリング
query_id文章ストリング
クエリ_メタデータ文章ストリング
スコアテンソル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 (デフォルト設定)

  • ダウンロードサイズ: 1.01 GiB

  • データセットサイズ: 4.53 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 8,841,823
'query' 509,962
'test' 9,139
'train' 516,472
'validation' 7,433

beir/trec_covid

  • ダウンロードサイズ: 70.45 MiB

  • データセットサイズ: 292.04 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 171,332
'query' 50
'test' 35,480

ビール/nfcorpus

  • ダウンロードサイズ: 2.34 MiB

  • データセットサイズ: 24.94 MiB

  • 自動キャッシュ(ドキュメント): はい

  • スプリット:

スプリット
'passage' 3,633
'query' 3,237
'test' 3,128
'train' 3,588
'validation' 3,119

ビール/nq

  • ダウンロードサイズ: 475.22 MiB

  • データセットサイズ: 1.66 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 2,681,468
'query' 3,452
'test' 4,201

ビール/ホットポットカ

  • ダウンロードサイズ: 623.73 MiB

  • データセットサイズ: 2.64 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 5,233,329
'query' 97,852
'test' 13,783
'train' 101,307
'validation' 10,335

ビール/フィーカ

  • ダウンロードサイズ: 17.12 MiB

  • データセットのサイズ: 73.39 MiB

  • 自動キャッシュ(ドキュメント): はい

  • スプリット:

スプリット
'passage' 57,638
'query' 6,648
'test' 1,706
'train' 14,166
'validation' 1,238

ビール/アルグアナ

  • ダウンロードサイズ: 3.60 MiB

  • データセットのサイズ: 15.08 MiB

  • 自動キャッシュ(ドキュメント): はい

  • スプリット:

スプリット
'passage' 8,674
'query' 1,406
'test' 1,401

beir/webis_touche2020

  • ダウンロードサイズ: 216.61 MiB

  • データセットのサイズ: 747.57 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 382,545
'query' 49
'test' 2,099

beir/cqadupstack.android

  • ダウンロードサイズ: 4.98 GiB

  • データセットのサイズ: 883.49 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 22,998
'query' 699
'test' 1,696

beir/cqadupstack.english

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 3.78 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 40,221
'query' 1,570
'test' 3,765

beir/cqadupstack.gaming

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 2.64 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 45,301
'query' 1,595
'test' 2,263

beir/cqadupstack.gis

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 1.47 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 37,637
'query' 885
'test' 1,114

beir/cqadupstack.mathematica

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 1.58 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 16,705
'query' 804
'test' 1,358

beir/cqadupstack.physics

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 2.02 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 38,316
'query' 1,039
'test' 1,933

beir/cqadupstack.programmers

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 2.14 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 32,176
'query' 876
'test' 1,675

beir/cqadupstack.stats

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 941.07 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 42,269
'query' 652
'test' 913

beir/cqadupstack.tex

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 21.75 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 68,184
'query' 2,906
'test' 5,154

beir/cqadupstack.unix

  • ダウンロードサイズ: 4.98 GiB

  • データセットサイズ: 2.04 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 47,382
'query' 1,072
'test' 1,693

beir/cqadupstack.webmasters

  • ダウンロードサイズ: 4.98 GiB

  • データセットのサイズ: 650.08 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 17,405
'query' 506
'test' 1,395

beir/cqadupstack.wordpress

  • ダウンロードサイズ: 4.98 GiB

  • データセットのサイズ: 726.15 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 48,605
'query' 541
'test' 744

ビール/クオラ

  • ダウンロードサイズ: 15.12 MiB

  • データセットサイズ: 121.33 MiB

  • 自動キャッシュ(ドキュメント): はい

  • スプリット:

スプリット
'passage' 522,931
'query' 15,000
'test' 15,675
'validation' 7,626

beir/dbpedia_entity

  • ダウンロードサイズ: 609.67 MiB

  • データセットサイズ: 2.47 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 4,635,922
'query' 467
'test' 40,724
'validation' 5,658

ベア/サイドックス

  • ダウンロードサイズ: 135.87 MiB

  • データセットのサイズ: 573.04 MiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 25,657
'query' 1,000
'test' 25,657

ビール/発熱

  • ダウンロードサイズ: 1.15 GiB

  • データセットサイズ: 3.70 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 5,416,568
'query' 123,142
'test' 1,499
'train' 12,547
'validation' 1,460

ビール/climate_fever

  • ダウンロードサイズ: 1.14 GiB

  • データセットサイズ: 3.64 GiB

  • 自動キャッシュ(ドキュメント): いいえ

  • スプリット:

スプリット
'passage' 5,416,593
'query' 1,535
'test' 1,344

ビール/サイファクト

  • ダウンロードサイズ: 2.69 MiB

  • データセットサイズ: 10.16 MiB

  • 自動キャッシュ(ドキュメント): はい

  • スプリット:

スプリット
'passage' 5,183
'query' 1,109
'test' 283
'train' 565