Skip to content

Commit

Permalink
feat: modify benchmark tasks
Browse files Browse the repository at this point in the history
  • Loading branch information
dbuades committed Nov 19, 2024
1 parent 701b892 commit 37ddab2
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions mteb/benchmarks/benchmarks.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,19 +309,24 @@ def load_results(
)

MTEB_RETRIEVAL_MEDICAL = Benchmark(
name="MTEB(Medical)",
name="medMTEB",
tasks=get_tasks(
tasks=[
"CUREv1",
"NFCorpus",
"SCIDOCS",
"TRECCOVID",
"TRECCOVID-PL",
"SciFact",
"SciFact-PL",
"MedicalQARetrieval",
"PublicHealthQA",
]
"MedrxivClusteringP2P.v2",
"MedrxivClusteringS2S.v2",
"CmedqaRetrieval",
"CMedQAv2-reranking",
],
),
description="Medical benchmarks from MTEB",
description="medMTEB: A curated set of MTEB tasks designed to evaluate systems in the context of medical information retrieval.",
reference="",
citation=None,
)
Expand Down Expand Up @@ -720,6 +725,7 @@ def load_results(
"SpartQA",
"TempReasonL1",
"TRECCOVID",
"CUREv1",
"WinoGrande",
"BelebeleRetrieval",
"MLQARetrieval",
Expand Down

0 comments on commit 37ddab2

Please sign in to comment.