regression.dl23-passage.splade-pp-ed.onnx.yaml Maven / Gradle / Ivy

Go to download
---
corpus: msmarco-v2-passage-splade-pp-ed
corpus_path: collections/msmarco/msmarco_v2_passage_splade_pp_ed/

download_url: https://rgw.cs.uwaterloo.ca/pyserini/data/msmarco_v2_passage_splade_pp_ed.tar
download_checksum: 2cdb2adc259b8fa6caf666b20ebdc0e8
download_corpus: msmarco_v2_passage_splade_pp_ed

index_path: indexes/lucene-inverted.msmarco-v2-passage.splade-pp-ed/
collection_class: JsonVectorCollection
generator_class: DefaultLuceneDocumentGenerator
index_threads: 24
index_options: -impact -pretokenized -storeRaw
index_stats:
  documents: 138364198
  documents (non-empty): 138364198
  total terms: 861058871141

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

topic_reader: TsvInt
topics:
  - name: "[DL23 (Passage)](https://microsoft.github.io/msmarco/TREC-Deep-Learning)"
    id: dl23
    path: topics.dl23.txt
    qrel: qrels.dl23-passage.txt

models:
  - name: splade-pp-ed-onnx
    display: SPLADE++ CoCondenser-EnsembleDistil
    params: -parallelism 16 -impact -pretokenized -encoder SpladePlusPlusEnsembleDistil
    results:
      MAP@100:
        - 0.1925
      MRR@100:
        - 0.6832
      nDCG@10:
        - 0.4727
      R@100:
        - 0.4137
      R@1000:
        - 0.6779
  - name: splade-pp-ed-onnx+rm3
    display: +RM3
    params: -parallelism 16 -impact -pretokenized -encoder SpladePlusPlusEnsembleDistil -rm3 -collection JsonVectorCollection
    results:
      MAP@100:
        - 0.1911
      MRR@100:
        - 0.5840
      nDCG@10:
        - 0.4409
      R@100:
        - 0.3979
      R@1000:
        - 0.6620
  - name: splade-pp-ed-onnx+rocchio
    display: +Rocchio
    params: -parallelism 16 -impact -pretokenized -encoder SpladePlusPlusEnsembleDistil -rocchio -collection JsonVectorCollection
    results:
      MAP@100:
        - 0.2014
      MRR@100:
        - 0.6843
      nDCG@10:
        - 0.4739
      R@100:
        - 0.4042
      R@1000:
        - 0.6934