-
Notifications
You must be signed in to change notification settings - Fork 38
Expand file tree
/
Copy pathconfig.yaml
More file actions
27 lines (19 loc) · 1.05 KB
/
config.yaml
File metadata and controls
27 lines (19 loc) · 1.05 KB
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
# Model ID (HuggingFace models).
model_id: "sentence-transformers/all-MiniLM-L6-v2"
# Accepted values: 'retrieval', 'reranking'
task_to_evaluate: "retrieval"
# Corpus jsonl file path with records - <id,title,text>.
corpus_path: "resources/mteb_datasets/nfcorpus/test/corpus.jsonl"
# Queries jsonl file path with records - <id,text>.
queries_path: "resources/mteb_datasets/nfcorpus/test/queries.jsonl"
# Candidates jsonl file path with records - <query_id,doc_id,rating>.
candidates_path: "resources/mteb_datasets/nfcorpus/test/candidates.jsonl"
# Relevance scale used in candidates dataset for rating field
# Accepted values:
# - binary: 0 (not relevant), 1 (relevant)
# - graded: 0 (not relevant), 1 (maybe ok), 2 (that’s my result)
relevance_scale: "graded"
# (Optional) Path to write mteb resources, if not given it will be written to resources dir in the root folder
output_dest: "resources"
# (Optional) Path to write mteb document and query embeddings, if not given it will be written to resources/embeddings dir
embeddings_dest: "resources/embeddings"