yiyanghkust commited on
Commit
1a5e5eb
1 Parent(s): 5c3a34e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -0
config.json CHANGED
@@ -1,6 +1,16 @@
1
  {"architectures": [
2
  "BertForSequenceClassification"
3
  ],
 
 
 
 
 
 
 
 
 
 
4
  "attention_probs_dropout_prob": 0.1,
5
  "hidden_act": "gelu",
6
  "hidden_dropout_prob": 0.1,
 
1
  {"architectures": [
2
  "BertForSequenceClassification"
3
  ],
4
+ "id2label": {
5
+ "0": "neutral",
6
+ "1": "positive",
7
+ "2": "negative"
8
+ },
9
+ "label2id": {
10
+ "positive": 1,
11
+ "negative": 2,
12
+ "neutral": 0
13
+ },
14
  "attention_probs_dropout_prob": 0.1,
15
  "hidden_act": "gelu",
16
  "hidden_dropout_prob": 0.1,