Align label mapping with imdb dataset

config.json

@@ -23,5 +23,13 @@
   "transformers_version": "4.10.2",
   "type_vocab_size": 2,
   "use_cache": true,
-  "vocab_size": 30522
-}
+  "vocab_size": 30522,
+  "label2id": {
+    "neg": 0,
+    "pos": 1
+  },
+  "id2label": {
+    "0": "neg",
+    "1": "pos"
+  }
+}