diff --git a/run_evaluation.py b/run_evaluation.py index b7ac173..54ebb97 100644 --- a/run_evaluation.py +++ b/run_evaluation.py @@ -95,7 +95,7 @@ # Create an empty text file to write results, and pass the name to eval temp_file = tempfile.NamedTemporaryFile(delete=False).name - EvalFunction.write_file(reranked_data, temp_file) + EvalFunction.write_file(new_results, temp_file) EvalFunction.main(THE_TOPICS[data], temp_file) @@ -127,5 +127,5 @@ from trec_eval import EvalFunction temp_file = tempfile.NamedTemporaryFile(delete=False).name - EvalFunction.write_file(reranked_data, temp_file) + EvalFunction.write_file(new_results, temp_file) EvalFunction.main(THE_TOPICS[data], temp_file)