Simplified Data Wrangling with ir_datasets
This tool simplifies data wrangling for IR researchers by providing a centralized resource, though it is incremental as it builds on existing dataset management needs.
The paper tackles the challenge of managing scattered documentation and diverse data formats in Information Retrieval (IR) experiments by introducing ir_datasets, a robust and lightweight tool for acquiring, managing, and operating over textual datasets for ad-hoc search, which integrates with popular IR toolkits and includes an extensive catalog.
Managing the data for Information Retrieval (IR) experiments can be challenging. Dataset documentation is scattered across the Internet and once one obtains a copy of the data, there are numerous different data formats to work with. Even basic formats can have subtle dataset-specific nuances that need to be considered for proper use. To help mitigate these challenges, we introduce a new robust and lightweight tool (ir_datasets) for acquiring, managing, and performing typical operations over datasets used in IR. We primarily focus on textual datasets used for ad-hoc search. This tool provides both a Python and command line interface to numerous IR datasets and benchmarks. To our knowledge, this is the most extensive tool of its kind. Integrations with popular IR indexing and experimentation toolkits demonstrate the tool's utility. We also provide documentation of these datasets through the ir_datasets catalog: https://ir-datasets.com/. The catalog acts as a hub for information on datasets used in IR, providing core information about what data each benchmark provides as well as links to more detailed information. We welcome community contributions and intend to continue to maintain and grow this tool.