dl20-passage-cos-dpr-distil-hnsw-onnx.yaml

---
corpus: msmarco-passage-cos-dpr-distil
corpus_path: collections/msmarco/msmarco-passage-cos-dpr-distil/

download_url: https://rgw.cs.uwaterloo.ca/pyserini/data/msmarco-passage-cos-dpr-distil.tar
download_checksum: e20ffbc8b5e7f760af31298aefeaebbd

index_path: indexes/lucene-hnsw.msmarco-passage-cos-dpr-distil/
index_type: hnsw
collection_class: JsonDenseVectorCollection
generator_class: HnswDenseVectorDocumentGenerator
index_threads: 16
index_options: -M 16 -efC 100 -memoryBuffer 65536 -noMerge

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

topic_reader: TsvInt
topics:
  - name: "[DL20 (Passage)](https://trec.nist.gov/data/deep2020.html)"
    id: dl20
    path: topics.dl20.txt
    qrel: qrels.dl20-passage.txt

models:
  - name: cos-dpr-distil-hnsw
    display: cosDPR-distil
    type: hnsw
    params: -generator VectorQueryGenerator -topicField title -threads 16 -hits 1000 -efSearch 1000 -encoder CosDprDistil
    results:
      AP@1000:
        - 0.482
      nDCG@10:
        - 0.701
      R@100:
        - 0.712
      R@1000:
        - 0.843