Hugging Face's logo

cirimus
/
modernbert-base-emotions

Text Classification
Transformers
Safetensors
PyTorch
English
modernbert
ModernBERT
emotions
multi-class-classification
Model card Files Community
cirimus commited on
Commit
64c9b38
·
verified ·
1 Parent(s): bfea7eb

First commit

Browse files
Files changed (1) hide show
  1. config.json +14 -14
config.json CHANGED
@@ -21,25 +21,25 @@
21
  "hidden_activation": "gelu",
22
  "hidden_size": 768,
23
  "id2label": {
24
- "0": "LABEL_0",
25
- "1": "LABEL_1",
26
- "2": "LABEL_2",
27
- "3": "LABEL_3",
28
- "4": "LABEL_4",
29
- "5": "LABEL_5",
30
- "6": "LABEL_6"
31
  },
32
  "initializer_cutoff_factor": 2.0,
33
  "initializer_range": 0.02,
34
  "intermediate_size": 1152,
35
  "label2id": {
36
- "LABEL_0": 0,
37
- "LABEL_1": 1,
38
- "LABEL_2": 2,
39
- "LABEL_3": 3,
40
- "LABEL_4": 4,
41
- "LABEL_5": 5,
42
- "LABEL_6": 6
43
  },
44
  "layer_norm_eps": 1e-05,
45
  "local_attention": 128,
 
21
  "hidden_activation": "gelu",
22
  "hidden_size": 768,
23
  "id2label": {
24
+ "0": "NEUTRAL",
25
+ "1": "SURPRISE",
26
+ "2": "FEAR",
27
+ "3": "SADNESS",
28
+ "4": "JOY",
29
+ "5": "ANGER",
30
+ "6": "LOVE"
31
  },
32
  "initializer_cutoff_factor": 2.0,
33
  "initializer_range": 0.02,
34
  "intermediate_size": 1152,
35
  "label2id": {
36
+ "ANGER": 5,
37
+ "FEAR": 2,
38
+ "JOY": 4,
39
+ "LOVE": 6,
40
+ "NEUTRAL": 0,
41
+ "SADNESS": 3,
42
+ "SURPRISE": 1
43
  },
44
  "layer_norm_eps": 1e-05,
45
  "local_attention": 128,