anserini/src/main/resources/reproduce/from-document-collection/configs/rag24-doc-segmented-test-umbrela.splade-v3.onnx.yaml at master · castorini/anserini · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
---
corpus: msmarco-v2.1-doc-segmented-splade-v3
corpus_path: collections/msmarco/msmarco_v2.1_doc_segmented_splade-v3

download_url: https://rgw.cs.uwaterloo.ca/pyserini/data/msmarco_v2.1_doc_segmented_splade-v3.tar
download_checksum: c62490569364a1eb0101da1ca4a894d9
download_corpus: msmarco_v2.1_doc_segmented_splade-v3

index_path: indexes/lucene-inverted.msmarco-v2.1-doc-segmented.splade-v3/
collection_class: JsonVectorCollection
generator_class: DefaultLuceneDocumentGenerator
index_threads: 24
index_options: -impact -pretokenized
index_stats:
  documents: 113520750
  documents (non-empty): 113520750
  total terms: 866904601378

metrics:
  - metric: nDCG@20
    command: bin/trec_eval
    params: -c -m ndcg_cut.20
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: false
  - metric: nDCG@100
    command: bin/trec_eval
    params: -c -m ndcg_cut.100
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: false
  - metric: R@100
    command: bin/trec_eval
    params: -c -m recall.100
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: false

topic_reader: TsvString
topics:
  - name: "RAG 24: Test queries"
    id: rag24.test
    path: topics.rag24.test.txt
    qrel: qrels.rag24.test-umbrela-all.txt

models:
  - name: splade-v3-onnx
    display: SPLADE-v3
    params: -impact -pretokenized -removeQuery -hits 1000 -encoder SpladeV3
    results:
      nDCG@20:
        - 0.5167
      nDCG@100:
        - 0.4587
      R@100:
        - 0.2437