anserini/src/main/resources/reproduce/from-document-collection/configs/bright-earth-science.bge-large-en-v1.5.flat.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
---
corpus: bright-earth-science.bge-large-en-v1.5
corpus_path: collections/bright/bge-large-en-v1.5/earth-science

index_path: indexes/lucene-flat.bright-earth-science.bge-large-en-v1.5/
index_type: flat
collection_class: JsonDenseVectorCollection
generator_class: DenseVectorDocumentGenerator
index_threads: 4
index_options: ""

metrics:
  - 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
    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: TsvString
topics:
  - name: "BRIGHT: Earth Science"
    id: topics
    path: topics.bright-earth-science.tsv.gz
    qrel: qrels.bright-earth-science.txt

models:
  - name: bge-flat-onnx
    display: BGE-large-en-v1.5
    type: flat
    params: -encoder BgeLargeEn15 -hits 1000 -removeQuery -threads 16
    results:
      nDCG@10:
        - 0.2545
      R@100:
        - 0.5399
      R@1000:
        - 0.7441