Incorporate additional explanations in dev/test sets.

PiperOrigin-RevId: 282425993

Author: sharannarang

tensorflow_datasets/testing/test_data/fake_examples/esnli/esnli_dev.csv

@@ -1,2 +1,2 @@
-pairID,gold_label,Sentence1,Sentence2,Explanation_1,WorkerId,Sentence1_marked_1,Sentence2_marked_1,Sentence1_Highlighted_1,Sentence2_Highlighted_1
-1d,entailment,A woman smiles at the child.,A woman is present.,A woman must be present to smile.,xyz,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3"
+pairID,gold_label,Sentence1,Sentence2,Explanation_1,Sentence1_marked_1,Sentence2_marked_1,Sentence1_Highlighted_1,Sentence2_Highlighted_1,Explanation_2,Sentence1_marked_2,Sentence2_marked_2,Sentence1_Highlighted_2,Sentence2_Highlighted_2,Explanation_3,Sentence1_marked_3,Sentence2_marked_3,Sentence1_Highlighted_3,Sentence2_Highlighted_3
+1d,entailment,A woman smiles at the child.,A woman is present.,A woman must be present to smile.,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3",A woman smiling implies that she is present.,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3",A smiling woman is also present.,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3"

tensorflow_datasets/testing/test_data/fake_examples/esnli/esnli_test.csv

@@ -1,2 +1,2 @@
-pairID,gold_label,Sentence1,Sentence2,Explanation_1,WorkerId,Sentence1_marked_1,Sentence2_marked_1,Sentence1_Highlighted_1,Sentence2_Highlighted_1
-1t,neutral,A woman looks at the child.,A woman is smiling.,Smiling and looking at the child are independent.,xyz,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3"
+pairID,gold_label,Sentence1,Sentence2,Explanation_1,Sentence1_marked_1,Sentence2_marked_1,Sentence1_Highlighted_1,Sentence2_Highlighted_1,Explanation_2,Sentence1_marked_2,Sentence2_marked_2,Sentence1_Highlighted_2,Sentence2_Highlighted_2,Explanation_3,Sentence1_marked_3,Sentence2_marked_3,Sentence1_Highlighted_3,Sentence2_Highlighted_3
+1t,neutral,A woman looks at the child.,A woman is smiling.,Smiling and looking at the child are independent.,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3",Smiling and looking at the child are independent.,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3",Smiling and looking at the child are independent.,A woman *smiles* at the child.,A woman is *present*,{},"1,2,3"

tensorflow_datasets/text/esnli.py

@@ -49,10 +49,14 @@
 class Esnli(tfds.core.GeneratorBasedBuilder):
   """e-SNLI: Natural Language Inference with Natural Language Explanations corpus."""
 
+  # Version History
+  # 0.0.2 Added explanation_2, explanation_3 fields which exist in the dev/test
+  # splits only.
+  # 0.0.1 Initial version
   BUILDER_CONFIGS = [
       tfds.core.BuilderConfig(
           name='plain_text',
-          version=tfds.core.Version('0.0.1'),
+          version=tfds.core.Version('0.0.2'),
           description='Plain text import of e-SNLI',
       )
   ]
@@ -69,7 +73,11 @@ def _info(self):
             'label':
                 tfds.features.ClassLabel(
                     names=['entailment', 'neutral', 'contradiction']),
-            'explanation':
+            'explanation_1':
+                tfds.features.Text(),
+            'explanation_2':
+                tfds.features.Text(),
+            'explanation_3':
                 tfds.features.Text(),
         }),
         supervised_keys=None,
@@ -112,5 +120,7 @@ def _generate_examples(self, files):
               'premise': row['Sentence1'],
               'hypothesis': row['Sentence2'],
               'label': row['gold_label'],
-              'explanation': row['Explanation_1']
+              'explanation_1': row['Explanation_1'],
+              'explanation_2': row.get('Explanation_2', ''),
+              'explanation_3': row.get('Explanation_3', ''),
           }