Update config.json

2021-10-15 06:27:31 +00:00 · 2021-10-15 06:27:31 +00:00 · 1a5e5eb5dd
parent 5c3a34ef12
commit 1a5e5eb5dd
1 changed files with 10 additions and 0 deletions
--- a/config.json
+++ b/config.json
@ -1,6 +1,16 @@
 {"architectures": [
    "BertForSequenceClassification"
  ],
+  "id2label": {
+    "0": "neutral",
+    "1": "positive",
+    "2": "negative"
+  },
+  "label2id": {
+    "positive": 1,
+    "negative": 2,
+    "neutral": 0
+  },
  "attention_probs_dropout_prob": 0.1,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.1,