- Description:
The WinoGrande, a large-scale dataset of 44k problems, inspired by the original Winograd Schema Challenge design, but adjusted to improve both the scale and the hardness of the dataset.
Additional Documentation: Explore on Papers With Code
Homepage: http://winogrande.allenai.org/
Source code:
tfds.text.Winogrande
Versions:
1.2.0
(default): Updated source file with more data and new checksums.
Download size:
3.24 MiB
Dataset size:
9.97 MiB
Auto-cached (documentation): Yes
Splits:
Split | Examples |
---|---|
'test' |
1,767 |
'train_l' |
10,234 |
'train_m' |
2,558 |
'train_s' |
640 |
'train_xl' |
40,398 |
'train_xs' |
160 |
'validation' |
1,267 |
- Feature structure:
FeaturesDict({
'label': ClassLabel(shape=(), dtype=int64, num_classes=2),
'option1': Text(shape=(), dtype=string),
'option2': Text(shape=(), dtype=string),
'sentence': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
FeaturesDict | ||||
label | ClassLabel | int64 | ||
option1 | Text | string | ||
option2 | Text | string | ||
sentence | Text | string |
Supervised keys (See
as_supervised
doc):None
Figure (tfds.show_examples): Not supported.
Examples (tfds.as_dataframe):
- Citation:
@article{sakaguchi2019winogrande,
title={WinoGrande: An Adversarial Winograd Schema Challenge at Scale},
author={Sakaguchi, Keisuke and Bras, Ronan Le and Bhagavatula, Chandra and Choi, Yejin},
journal={arXiv preprint arXiv:1907.10641},
year={2019}
}