TensorFlow 2.0 RC is available Learn more

snli

The SNLI corpus (version 1.0) is a collection of 570k human-written English sentence pairs manually labeled for balanced classification with the labels entailment, contradiction, and neutral, supporting the task of natural language inference (NLI), also known as recognizing textual entailment (RTE).

snli is configured with tfds.text.snli.BuilderConfig and has the following configurations predefined (defaults to the first one):

  • plain_text (v0.0.1) (Size: 90.17 MiB): Plain text import of SNLI

snli/plain_text

FeaturesDict({
    'hypothesis': Text(shape=(), dtype=tf.string),
    'label': ClassLabel(shape=(), dtype=tf.int64, num_classes=3),
    'premise': Text(shape=(), dtype=tf.string),
})

Statistics

Split Examples
ALL 570,152
TRAIN 550,152
VALIDATION 10,000
TEST 10,000

Urls

Supervised keys (for as_supervised=True)

None

Citation

@inproceedings{snli:emnlp2015,
    Author = {Bowman, Samuel R. and Angeli, Gabor and Potts, Christopher, and Manning, Christopher D.},
    Booktitle = {Proceedings of the 2015 Conference on Empirical Methods in Natural Language Processing (EMNLP)},
    Publisher = {Association for Computational Linguistics},
    Title = {A large annotated corpus for learning natural language inference},
    Year = {2015}
}