jasmeeetsingh commited on
Commit
46ca71d
1 Parent(s): 754d6b9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -5
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "cardiffnlp/twitter-roberta-base-mar2022",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
@@ -13,15 +13,13 @@
13
  "hidden_size": 768,
14
  "id2label": {
15
  "0": "not depression",
16
- "1": "moderate",
17
- "2": "severe"
18
  },
19
  "initializer_range": 0.02,
20
  "intermediate_size": 3072,
21
  "label2id": {
22
  "not depression": 0,
23
- "moderate": 1,
24
- "severe": 2
25
  },
26
  "layer_norm_eps": 1e-05,
27
  "max_position_embeddings": 514,
 
1
  {
2
+ "_name_or_path": "jasmeeetsingh/twitter-depression-classification-sentiment140",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
 
13
  "hidden_size": 768,
14
  "id2label": {
15
  "0": "not depression",
16
+ "1": "moderate"
 
17
  },
18
  "initializer_range": 0.02,
19
  "intermediate_size": 3072,
20
  "label2id": {
21
  "not depression": 0,
22
+ "moderate": 1
 
23
  },
24
  "layer_norm_eps": 1e-05,
25
  "max_position_embeddings": 514,