from hamu_tool.dataset import DataLoader
loader = DataLoader.load('beir/trec-covid')
for doc in loader.get_docs():
print(doc.id, doc.text, doc.title, doc.url, doc.pubmed_id)
break
for query in loader.get_queries():
print(query.id, query.text, query.narrative)
break
for qrel in loader.get_qrels('[mode]'):
print(qrel.qid, qrel.did, qrel.score)
break
@article{Wang2020Cord19,
title = "CORD-19: The Covid-19 Open Research Dataset",
author = "Lucy Lu Wang and Kyle Lo and Yoganand Chandrasekhar and Russell Reas and Jiangjiang Yang and Darrin Eide and K. Funk and Rodney Michael Kinney and Ziyang Liu and W. Merrill and P. Mooney and D. Murdick and Devvret Rishi and Jerry Sheehan and Zhihong Shen and B. Stilson and A. Wade and K. Wang and Christopher Wilhelm and Boya Xie and D. Raymond and Daniel S. Weld and Oren Etzioni and Sebastian Kohlmeier",
journal = "ArXiv",
year = "2020"
}
@article{Voorhees2020TrecCovid,
title = "TREC-COVID: Constructing a Pandemic Information Retrieval Test Collection",
author = "E. Voorhees and Tasmeer Alam and Steven Bedrick and Dina Demner-Fushman and W. Hersh and Kyle Lo and Kirk Roberts and I. Soboroff and Lucy Lu Wang",
journal = "ArXiv",
year = "2020",
volume = "abs/2005.04474"
}
@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",
}