regression.beir-v1.0.0-webis-touche2020.bge-base-en-v1.5.parquet.flat-int8.onnx.yaml Maven / Gradle / Ivy

Go to download
---
corpus: beir-v1.0.0-webis-touche2020.bge-base-en-v1.5
corpus_path: collections/beir-v1.0.0/bge-base-en-v1.5/webis-touche2020.parquet

index_path: indexes/lucene-flat-int8.beir-v1.0.0-webis-touche2020.bge-base-en-v1.5/
index_type: flat
collection_class: ParquetDenseVectorCollection
generator_class: DenseVectorDocumentGenerator
index_threads: 16
index_options: -quantize.int8

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: "BEIR (v1.0.0): Webis-Touche2020"
    id: test
    path: topics.beir-v1.0.0-webis-touche2020.test.tsv.gz
    qrel: qrels.beir-v1.0.0-webis-touche2020.test.txt

models:
  - name: bge-flat-int8-onnx
    display: BGE-base-en-v1.5
    type: flat
    params: -encoder BgeBaseEn15 -hits 1000 -removeQuery -threads 16
    results:
      nDCG@10:
        - 0.2570
      R@100:
        - 0.4857
      R@1000:
        - 0.8298
    tolerance:
      nDCG@10:
        - 0.008
      R@100:
        - 0.001
      R@1000:
        - 0.005