- Description:
Cosmos QA is a large-scale dataset of 35.6K problems that require commonsense-based reading comprehension, formulated as multiple-choice questions. It focuses on reading between the lines over a diverse collection of people's everyday narratives, asking questions concerning on the likely causes or effects of events that require reasoning beyond the exact text spans in the context.
Additional Documentation: Explore on Papers With Code
Homepage: https://wilburone.github.io/cosmos/
Source code:
tfds.question_answering.CosmosQA
Versions:
1.0.0
(default): No release notes.
Download size:
23.27 MiB
Dataset size:
27.09 MiB
Auto-cached (documentation): Yes
Splits:
Split | Examples |
---|---|
'test' |
6,963 |
'train' |
25,262 |
'validation' |
2,985 |
- Feature structure:
FeaturesDict({
'answer0': Text(shape=(), dtype=string),
'answer1': Text(shape=(), dtype=string),
'answer2': Text(shape=(), dtype=string),
'answer3': Text(shape=(), dtype=string),
'context': Text(shape=(), dtype=string),
'id': Text(shape=(), dtype=string),
'label': ClassLabel(shape=(), dtype=int64, num_classes=4),
'question': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
FeaturesDict | ||||
answer0 | Text | string | ||
answer1 | Text | string | ||
answer2 | Text | string | ||
answer3 | Text | string | ||
context | Text | string | ||
id | Text | string | ||
label | ClassLabel | int64 | ||
question | Text | string |
Supervised keys (See
as_supervised
doc):None
Figure (tfds.show_examples): Not supported.
Examples (tfds.as_dataframe):
- Citation:
@inproceedings{huang-etal-2019-cosmos,
title = "Cosmos {QA}: Machine Reading Comprehension with Contextual Commonsense Reasoning",
author = "Huang, Lifu and
Le Bras, Ronan and
Bhagavatula, Chandra and
Choi, Yejin",
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP)",
year = "2019",
url = "https://www.aclweb.org/anthology/D19-1243"
}