cirimus commited on
Commit
c1cc11c
·
verified ·
1 Parent(s): 4be6d5e

Add model with label mapping

Browse files
Files changed (1) hide show
  1. config.json +57 -57
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "./finetuned_model",
3
  "architectures": [
4
  "ModernBertForSequenceClassification"
5
  ],
@@ -21,67 +21,67 @@
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
- "7": "LABEL_7",
32
- "8": "LABEL_8",
33
- "9": "LABEL_9",
34
- "10": "LABEL_10",
35
- "11": "LABEL_11",
36
- "12": "LABEL_12",
37
- "13": "LABEL_13",
38
- "14": "LABEL_14",
39
- "15": "LABEL_15",
40
- "16": "LABEL_16",
41
- "17": "LABEL_17",
42
- "18": "LABEL_18",
43
- "19": "LABEL_19",
44
- "20": "LABEL_20",
45
- "21": "LABEL_21",
46
- "22": "LABEL_22",
47
- "23": "LABEL_23",
48
- "24": "LABEL_24",
49
- "25": "LABEL_25",
50
- "26": "LABEL_26",
51
- "27": "LABEL_27"
52
  },
53
  "initializer_cutoff_factor": 2.0,
54
  "initializer_range": 0.02,
55
  "intermediate_size": 1152,
56
  "label2id": {
57
- "LABEL_0": 0,
58
- "LABEL_1": 1,
59
- "LABEL_10": 10,
60
- "LABEL_11": 11,
61
- "LABEL_12": 12,
62
- "LABEL_13": 13,
63
- "LABEL_14": 14,
64
- "LABEL_15": 15,
65
- "LABEL_16": 16,
66
- "LABEL_17": 17,
67
- "LABEL_18": 18,
68
- "LABEL_19": 19,
69
- "LABEL_2": 2,
70
- "LABEL_20": 20,
71
- "LABEL_21": 21,
72
- "LABEL_22": 22,
73
- "LABEL_23": 23,
74
- "LABEL_24": 24,
75
- "LABEL_25": 25,
76
- "LABEL_26": 26,
77
- "LABEL_27": 27,
78
- "LABEL_3": 3,
79
- "LABEL_4": 4,
80
- "LABEL_5": 5,
81
- "LABEL_6": 6,
82
- "LABEL_7": 7,
83
- "LABEL_8": 8,
84
- "LABEL_9": 9
85
  },
86
  "layer_norm_eps": 1e-05,
87
  "local_attention": 128,
 
1
  {
2
+ "_name_or_path": "./modernbert-base-go-emotions",
3
  "architectures": [
4
  "ModernBertForSequenceClassification"
5
  ],
 
21
  "hidden_activation": "gelu",
22
  "hidden_size": 768,
23
  "id2label": {
24
+ "0": "admiration",
25
+ "1": "amusement",
26
+ "2": "anger",
27
+ "3": "annoyance",
28
+ "4": "approval",
29
+ "5": "caring",
30
+ "6": "confusion",
31
+ "7": "curiosity",
32
+ "8": "desire",
33
+ "9": "disappointment",
34
+ "10": "disapproval",
35
+ "11": "disgust",
36
+ "12": "embarrassment",
37
+ "13": "excitement",
38
+ "14": "fear",
39
+ "15": "gratitude",
40
+ "16": "grief",
41
+ "17": "joy",
42
+ "18": "love",
43
+ "19": "nervousness",
44
+ "20": "optimism",
45
+ "21": "pride",
46
+ "22": "realization",
47
+ "23": "relief",
48
+ "24": "remorse",
49
+ "25": "sadness",
50
+ "26": "surprise",
51
+ "27": "neutral"
52
  },
53
  "initializer_cutoff_factor": 2.0,
54
  "initializer_range": 0.02,
55
  "intermediate_size": 1152,
56
  "label2id": {
57
+ "admiration": 0,
58
+ "amusement": 1,
59
+ "anger": 2,
60
+ "annoyance": 3,
61
+ "approval": 4,
62
+ "caring": 5,
63
+ "confusion": 6,
64
+ "curiosity": 7,
65
+ "desire": 8,
66
+ "disappointment": 9,
67
+ "disapproval": 10,
68
+ "disgust": 11,
69
+ "embarrassment": 12,
70
+ "excitement": 13,
71
+ "fear": 14,
72
+ "gratitude": 15,
73
+ "grief": 16,
74
+ "joy": 17,
75
+ "love": 18,
76
+ "nervousness": 19,
77
+ "neutral": 27,
78
+ "optimism": 20,
79
+ "pride": 21,
80
+ "realization": 22,
81
+ "relief": 23,
82
+ "remorse": 24,
83
+ "sadness": 25,
84
+ "surprise": 26
85
  },
86
  "layer_norm_eps": 1e-05,
87
  "local_attention": 128,