Update config.json

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/home/zengrui/zengrui/nlp_backdoor_models/combination-lock-yelp/backdoor-bert-base-uncased-poison-ratio-10-percent-target-0-model-id-4",
+  "_name_or_path": "JiaqiLee/robust-bert-yelp",
   "architectures": [
     "BertForSequenceClassification"
   ],
@@ -18,10 +18,17 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "return_dict": false,
   "torch_dtype": "float32",
   "transformers_version": "4.11.3",
   "type_vocab_size": 2,
   "use_cache": true,
-  "vocab_size": 30522
+  "vocab_size": 30522,
+  "id2label": {
+    "0": "negative",
+    "1": "positive"
+  },
+  "label2id": {
+    "negative": 0,
+    "positive": 1
+  }
 }