Commit a8eb74d7 authored by Vít Novotný's avatar Vít Novotný
Browse files

Shorten a line in scripts.common.ndcg

parent 0e794f16
Pipeline #57622 failed with stage
in 41 seconds
......@@ -33,9 +33,5 @@ def ndcg(parsed_run, task='task1', subset='train'):
evaluator = EVALUATORS[subset][task]
only_judged_parsed_run = remove_nonjudged_topics_and_documents(parsed_run, task, subset)
evaluation = evaluator.evaluate(only_judged_parsed_run)
ndcg = np.mean([
for topic, measures
in evaluation.items()
ndcg = np.mean([measures['ndcg'] for topic, measures in evaluation.items()])
return ndcg
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment