regression.msmarco-v1-passage.openai-ada2.parquet.hnsw-int8.cached.yaml Maven / Gradle / Ivy

Go to download
---
corpus: msmarco-passage-openai-ada2.parquet
corpus_path: collections/msmarco/msmarco-passage-openai-ada2.parquet/

download_url: https://rgw.cs.uwaterloo.ca/pyserini/data/msmarco-passage-openai-ada2.parquet.tar
download_checksum: a8fddf594c9b8e771637968033b12f6d

index_path: indexes/lucene-hnsw-int8.msmarco-v1-passage.openai-ada2/
index_type: hnsw
collection_class: ParquetDenseVectorCollection
generator_class: DenseVectorDocumentGenerator
index_threads: 16
index_options: -M 16 -efC 100 -quantize.int8

metrics:
  - metric: AP@1000
    command: bin/trec_eval
    params: -c -m map
    separator: "\t"
    parse_index: 2
    metric_precision: 4
    can_combine: false
  - metric: RR@10
    command: bin/trec_eval
    params: -c -M 10 -m recip_rank
    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: JsonIntVector
topics:
  - name: "[MS MARCO Passage: Dev](https://github.com/microsoft/MSMARCO-Passage-Ranking)"
    id: dev
    path: topics.msmarco-passage.dev-subset.openai-ada2.jsonl.gz
    qrel: qrels.msmarco-passage.dev-subset.txt

models:
  - name: openai-ada2-hnsw-int8-cached
    display: OpenAI-ada2
    type: hnsw
    params: -hits 1000 -efSearch 1000 -threads 16
    results:
      AP@1000:
        - 0.3505
      RR@10:
        - 0.3434
      R@100:
        - 0.8996
      R@1000:
        - 0.9858
    tolerance:
      AP@1000:
        - 0.015
      RR@10:
        - 0.015
      R@100:
        - 0.01
      R@1000:
        - 0.006