Update config.json

2019-12-20 12:05:06 +00:00 · 2019-12-20 12:05:06 +00:00 · 32d9c76413
parent dc58a97dfe
commit 32d9c76413
1 changed files with 9 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -21,5 +21,13 @@
  "tie_weights_": true,
  "torchscript": false,
  "use_bfloat16": false,
-  "vocab_size": 30522
+  "vocab_size": 30522,
+  "label2id": {
+    "NEGATIVE": 0,
+    "POSITIVE": 1
+  },
+  "id2label": {
+    "0": "NEGATIVE",
+    "1": "POSITIVE"
+  }
 }