Skip to content

Commit

Permalink
added 40 to k of ndcg
Browse files Browse the repository at this point in the history
  • Loading branch information
masaya-ohagi committed Apr 4, 2024
1 parent ca9c6ae commit 5f8a183
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/jmteb/evaluators/reranking/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def __init__(
) -> None:
self.query_dataset = query_dataset
self.doc_dataset = doc_dataset
self.ndcg_at_k = ndcg_at_k or [10, 20]
self.ndcg_at_k = ndcg_at_k or [10, 20, 40]
self.main_metric = f"ndcg@{self.ndcg_at_k[0]}"

def __call__(
Expand Down

0 comments on commit 5f8a183

Please sign in to comment.