barissayil
/

bert-sentiment-analysis-sst

Text Classification

Inference Endpoints

Model card Files Files and versions Community

LysandreJik commited on Jun 11, 2021

Commit

91b0621

•

1 Parent(s): 1955f23

Update

Files changed (1) hide show

config.json +6 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "SentimentClassifier"
   ],
   "attention_probs_dropout_prob": 0.1,
   "gradient_checkpointing": false,
@@ -16,5 +16,9 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "type_vocab_size": 2,
-  "vocab_size": 30522
 }

 {
   "architectures": [
+    "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "gradient_checkpointing": false,
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "type_vocab_size": 2,
+  "vocab_size": 30522,
+  "id2label": {
+    "0": "NEGATIVE",
+    "1": "POSITIVE"
+  }
 }