regression.msmarco-v2-passage.d2q-t5.yaml Maven / Gradle / Ivy

Go to download
---
corpus: msmarco-v2-passage-d2q-t5
corpus_path: collections/msmarco/msmarco_v2_passage_d2q-t5/

index_path: indexes/lucene-inverted.msmarco-v2-passage.d2q-t5/
collection_class: MsMarcoV2PassageCollection
generator_class: DefaultLuceneDocumentGenerator
index_threads: 24
index_options: -storeRaw
index_stats:
  documents: 138364198
  documents (non-empty): 138364198
  total terms: 16961479264

metrics:
  - metric: MAP@100
    command: bin/trec_eval
    params: -c -M 100 -m map
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: true
  - metric: MRR@100
    command: bin/trec_eval
    params: -c -M 100 -m recip_rank
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: true
  - metric: R@100
    command: bin/trec_eval
    params: -c -m recall.100
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: false
  - metric: R@1000
    command: bin/trec_eval
    params: -c -m recall.1000
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: false

topic_reader: TsvInt
topics:
  - name: "[MS MARCO V2 Passage: Dev](https://microsoft.github.io/msmarco/TREC-Deep-Learning.html)"
    id: dev
    path: topics.msmarco-v2-passage.dev.txt
    qrel: qrels.msmarco-v2-passage.dev.txt
  - name: "[MS MARCO V2 Passage: Dev2](https://microsoft.github.io/msmarco/TREC-Deep-Learning.html)"
    id: dev2
    path: topics.msmarco-v2-passage.dev2.txt
    qrel: qrels.msmarco-v2-passage.dev2.txt

models:
  - name: bm25-default
    display: BM25 (default)
    params: -bm25
    results:
      MAP@100:
        - 0.1057
        - 0.1112
      MRR@100:
        - 0.1072
        - 0.1123
      R@100:
        - 0.4670
        - 0.4803
      R@1000:
        - 0.7083
        - 0.7151
# PRF regressions are no longer maintained for sparse judgments to reduce running times.
# (commenting out instead of removing; in case these numbers are needed, just uncomment and rerun.)
#
#  - name: bm25-default+rm3
#    display: +RM3
#    params: -bm25 -rm3 -collection MsMarcoV2PassageCollection
#    results:
#      MAP@100:
#        - 0.0938
#        - 0.0976
#      MRR@100:
#        - 0.0947
#        - 0.0984
#      R@100:
#        - 0.4713
#        - 0.4803
#      R@1000:
#        - 0.7181
#        - 0.7222
#  - name: bm25-default+rocchio
#    display: +Rocchio
#    params: -bm25 -rocchio -collection MsMarcoV2PassageCollection
#    results:
#      MAP@100:
#        - 0.0943
#        - 0.0990
#      MRR@100:
#        - 0.0952
#        - 0.0997
#      R@100:
#        - 0.4723
#        - 0.4815
#      R@1000:
#        - 0.7211
#        - 0.7274