Update config.json

2021-10-15 06:11:05 +00:00 · 2021-10-15 06:11:05 +00:00 · b68b7aac74
parent f1c1f5a095
commit b68b7aac74
1 changed files with 9 additions and 3 deletions
--- a/config.json
+++ b/config.json
@ -3,10 +3,16 @@
    "BertForSequenceClassification"
  ],
  "id2label": {
-    "0": "positive",
-    "1": "negative",
-    "2": "neutral"
+    "0": "neutral",
+    "1": "positive",
+    "2": "negative"
  },
+  "label2id": {
+    "positive": 1,
+    "negative": 2,
+    "neutral": 0
+  },
+  "model_type": "bert",
  "attention_probs_dropout_prob": 0.1,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.1,