regression.dl21-doc-segmented.yaml Maven / Gradle / Ivy

Go to download
---
corpus: msmarco-v2-doc-segmented
corpus_path: collections/msmarco/msmarco_v2_doc_segmented/

index_path: indexes/lucene-index.msmarco-v2-doc-segmented/
collection_class: MsMarcoV2DocCollection
generator_class: DefaultLuceneDocumentGenerator
index_threads: 18
index_options: -storePositions -storeDocvectors -storeRaw
index_stats:
  documents: 124131414
  documents (non-empty): 124131414
  total terms: 24780918039

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

topic_reader: TsvInt
topic_root: src/main/resources/topics-and-qrels/
qrels_root: src/main/resources/topics-and-qrels/
topics:
  - name: "[DL21 (Doc)](https://microsoft.github.io/msmarco/TREC-Deep-Learning)"
    id: dl21
    path: topics.dl21.txt
    qrel: qrels.dl21-doc.txt

models:
  - name: bm25-default
    display: BM25 (default)
    params: -hits 10000 -selectMaxPassage -selectMaxPassage.delimiter "#" -selectMaxPassage.hits 1000 -bm25
    results:
      MAP@100:
        - 0.2436
      MRR@100:
        - 0.8937
      nDCG@10:
        - 0.5776
      R@100:
        - 0.3478
      R@1000:
        - 0.6930
  - name: bm25-default+rm3
    display: +RM3
    params: -hits 10000 -selectMaxPassage -selectMaxPassage.delimiter "#" -selectMaxPassage.hits 1000 -bm25 -rm3
    results:
      MAP@100:
        - 0.2936
      MRR@100:
        - 0.9076
      nDCG@10:
        - 0.6189
      R@100:
        - 0.3890
      R@1000:
        - 0.7678
  - name: bm25-default+rocchio
    display: +Rocchio
    params: -hits 10000 -selectMaxPassage -selectMaxPassage.delimiter "#" -selectMaxPassage.hits 1000 -bm25 -rocchio
    results:
      MAP@100:
        - 0.2974
      MRR@100:
        - 0.9099
      nDCG@10:
        - 0.6067
      R@100:
        - 0.3957
      R@1000:
        - 0.7740