def precision(e: trectools.TrecEval, per_query=False): rel_ret = e.get_relevant_retrieved_documents(per_query=per_query) rel = e.get_retrieved_documents(per_query=per_query) if per_query: return (rel_ret / rel).fillna(0) else: return rel_ret / rel
def recall(e: trectools.TrecEval) -> float: return e.get_relevant_retrieved_documents(per_query=False) / e.get_relevant_documents(per_query=False)