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

Shorten a line in scripts.common.ndcg

parent 0e794f16

scripts/common.py

+1 −5
Original line number Diff line number Diff line
@@ -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([

        measures['ndcg']

        for topic, measures

        in evaluation.items()

    ])

    ndcg = np.mean([measures['ndcg'] for topic, measures in evaluation.items()])

    return ndcg