abhiramd22
/

finetuning-sentiment-model-mpnet-imdb

Text Classification

generated_from_trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

abhiramd22 commited on Apr 2

Commit

feb9cad

•

1 Parent(s): 00d4855

Update config.json

Files changed (1) hide show

config.json +9 -1

config.json CHANGED Viewed

@@ -21,5 +21,13 @@
   "relative_attention_num_buckets": 32,
   "torch_dtype": "float32",
   "transformers_version": "4.38.2",
-  "vocab_size": 30527
 }

   "relative_attention_num_buckets": 32,
   "torch_dtype": "float32",
   "transformers_version": "4.38.2",
+  "vocab_size": 30527,
+  "label2id": {
+    "NEGATIVE": "0",
+    "POSITIVE": "1"
+  },
+  "id2label": {
+    "0": "NEGATIVE",
+    "1": "POSITIVE"
+  }
 }