from hamu_tool.dataset import DataLoader
loader = DataLoader.load('beir/robust04')
for doc in loader.get_docs():
print(doc.id, doc.text)
break
for query in loader.get_queries():
print(query.id, query.text)
break
for qrel in loader.get_qrels('[mode]'):
print(qrel.qid, qrel.did, qrel.score)
break
inproceedings{Voorhees2004Robust,
title = "Overview of the TREC 2004 Robust Retrieval Track",
author = "Ellen Voorhees",
booktitle = "TREC}",
year = "2004"
}
@article{Thakur2021Beir,
title = "BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Models",
author = "Thakur, Nandan and Reimers, Nils and Rücklé, Andreas and Srivastava, Abhishek and Gurevych, Iryna",
journal = "arXiv preprint arXiv:2104.08663",
month = "4",
year = "2021",
url = "https://arxiv.org/abs/2104.08663",
}