capreolus.utils.trec

Module Contents

Functions

threshold_trec_run(run, fold, k)

Take a trec run, and keep only the top-k docs

load_ntcir_topics(fn)

load_trec_topics(queryfn)

load_qrels(qrelfile[, qids, include_spam])

write_qrels(labels, qrelfile)

document_to_trectxt(docno, txt)

topic_to_trectxt(qno, title[, desc, narr])

anserini_index_to_trec_docs(index_dir, output_dir, ...)

capreolus.utils.trec.threshold_trec_run(run, fold, k)[source]

Take a trec run, and keep only the top-k docs

capreolus.utils.trec.load_ntcir_topics(fn)[source]
capreolus.utils.trec.load_trec_topics(queryfn)[source]
capreolus.utils.trec.load_qrels(qrelfile, qids=None, include_spam=True)[source]
capreolus.utils.trec.write_qrels(labels, qrelfile)[source]
capreolus.utils.trec.document_to_trectxt(docno, txt)[source]
capreolus.utils.trec.topic_to_trectxt(qno, title, desc=None, narr=None)[source]
capreolus.utils.trec.anserini_index_to_trec_docs(index_dir, output_dir, expected_doc_count)[source]