Hugging Face's logo

mkrausio
/
bert-emotion-ordinal

Transformers
Safetensors
bert-emotion-ordinal-regression
Generated from Trainer
Inference Endpoints
Model card Files Community
mkrausio commited on
Commit
37186bb
·
verified ·
1 Parent(s): 3ce6069

Upload BertForMultiOutputOrdinalRegression

Browse files
Files changed (1) hide show
  1. config.json +289 -19
config.json CHANGED
@@ -1,26 +1,296 @@
1
  {
2
- "_name_or_path": "bert-base-uncased",
3
  "architectures": [
4
  "BertForMultiOutputOrdinalRegression"
5
  ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "classifier_dropout": null,
8
- "gradient_checkpointing": false,
9
- "hidden_act": "gelu",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  "hidden_dropout_prob": 0.1,
11
- "hidden_size": 768,
12
- "initializer_range": 0.02,
13
- "intermediate_size": 3072,
14
- "layer_norm_eps": 1e-12,
15
- "max_position_embeddings": 512,
16
- "model_type": "bert",
17
- "num_attention_heads": 12,
18
- "num_hidden_layers": 12,
19
- "pad_token_id": 0,
20
- "position_embedding_type": "absolute",
21
  "torch_dtype": "float32",
22
- "transformers_version": "4.48.3",
23
- "type_vocab_size": 2,
24
- "use_cache": true,
25
- "vocab_size": 30522
26
  }
 
1
  {
 
2
  "architectures": [
3
  "BertForMultiOutputOrdinalRegression"
4
  ],
5
+ "bert_config": {
6
+ "_attn_implementation_autoset": true,
7
+ "architectures": [
8
+ "BertForMaskedLM"
9
+ ],
10
+ "gradient_checkpointing": false,
11
+ "model_type": "bert"
12
+ },
13
+ "categories_mapping": [
14
+ [
15
+ "Amusement",
16
+ 0,
17
+ 4
18
+ ],
19
+ [
20
+ "Elation",
21
+ 0,
22
+ 4
23
+ ],
24
+ [
25
+ "Pleasure/Ecstasy",
26
+ 0,
27
+ 4
28
+ ],
29
+ [
30
+ "Contentment",
31
+ 0,
32
+ 4
33
+ ],
34
+ [
35
+ "Thankfulness/Gratitude",
36
+ 0,
37
+ 4
38
+ ],
39
+ [
40
+ "Affection",
41
+ 0,
42
+ 4
43
+ ],
44
+ [
45
+ "Infatuation",
46
+ 0,
47
+ 4
48
+ ],
49
+ [
50
+ "Hope/Enthusiasm/Optimism",
51
+ 0,
52
+ 4
53
+ ],
54
+ [
55
+ "Triumph",
56
+ 0,
57
+ 4
58
+ ],
59
+ [
60
+ "Pride",
61
+ 0,
62
+ 4
63
+ ],
64
+ [
65
+ "Interest",
66
+ 0,
67
+ 4
68
+ ],
69
+ [
70
+ "Awe",
71
+ 0,
72
+ 4
73
+ ],
74
+ [
75
+ "Astonishment/Surprise",
76
+ 0,
77
+ 4
78
+ ],
79
+ [
80
+ "Concentration",
81
+ 0,
82
+ 4
83
+ ],
84
+ [
85
+ "Contemplation",
86
+ 0,
87
+ 4
88
+ ],
89
+ [
90
+ "Relief",
91
+ 0,
92
+ 4
93
+ ],
94
+ [
95
+ "Longing",
96
+ 0,
97
+ 4
98
+ ],
99
+ [
100
+ "Teasing",
101
+ 0,
102
+ 4
103
+ ],
104
+ [
105
+ "Impatience and Irritability",
106
+ 0,
107
+ 4
108
+ ],
109
+ [
110
+ "Sexual Lust",
111
+ 0,
112
+ 4
113
+ ],
114
+ [
115
+ "Doubt",
116
+ 0,
117
+ 4
118
+ ],
119
+ [
120
+ "Fear",
121
+ 0,
122
+ 4
123
+ ],
124
+ [
125
+ "Distress",
126
+ 0,
127
+ 4
128
+ ],
129
+ [
130
+ "Confusion",
131
+ 0,
132
+ 4
133
+ ],
134
+ [
135
+ "Embarrassment",
136
+ 0,
137
+ 4
138
+ ],
139
+ [
140
+ "Shame",
141
+ 0,
142
+ 4
143
+ ],
144
+ [
145
+ "Disappointment",
146
+ 0,
147
+ 4
148
+ ],
149
+ [
150
+ "Sadness",
151
+ 0,
152
+ 4
153
+ ],
154
+ [
155
+ "Bitterness",
156
+ 0,
157
+ 4
158
+ ],
159
+ [
160
+ "Contempt",
161
+ 0,
162
+ 4
163
+ ],
164
+ [
165
+ "Disgust",
166
+ 0,
167
+ 4
168
+ ],
169
+ [
170
+ "Anger",
171
+ 0,
172
+ 4
173
+ ],
174
+ [
175
+ "Malevolence/Malice",
176
+ 0,
177
+ 4
178
+ ],
179
+ [
180
+ "Sourness",
181
+ 0,
182
+ 4
183
+ ],
184
+ [
185
+ "Pain",
186
+ 0,
187
+ 4
188
+ ],
189
+ [
190
+ "Helplessness",
191
+ 0,
192
+ 4
193
+ ],
194
+ [
195
+ "Fatigue/Exhaustion",
196
+ 0,
197
+ 4
198
+ ],
199
+ [
200
+ "Emotional Numbness",
201
+ 0,
202
+ 4
203
+ ],
204
+ [
205
+ "Intoxication/Altered States of Consciousness",
206
+ 0,
207
+ 4
208
+ ],
209
+ [
210
+ "Jealousy & Envy",
211
+ 0,
212
+ 4
213
+ ]
214
+ ],
215
+ "dimensions_mapping": [
216
+ [
217
+ "Valence",
218
+ -3.0,
219
+ 3.0
220
+ ],
221
+ [
222
+ "Arousal",
223
+ 0.0,
224
+ 4.0
225
+ ],
226
+ [
227
+ "Submissive vs. Dominant",
228
+ -3.0,
229
+ 3.0
230
+ ],
231
+ [
232
+ "Age",
233
+ 0.0,
234
+ 6.0
235
+ ],
236
+ [
237
+ "Gender",
238
+ -2.0,
239
+ 2.0
240
+ ],
241
+ [
242
+ "Serious vs. Humorous",
243
+ 0.0,
244
+ 4.0
245
+ ],
246
+ [
247
+ "Vulnerable vs. Emotionally Detached",
248
+ 0.0,
249
+ 4.0
250
+ ],
251
+ [
252
+ "Confident vs. Hesitant",
253
+ 0.0,
254
+ 4.0
255
+ ],
256
+ [
257
+ "Warm vs. Cold",
258
+ -2.0,
259
+ 2.0
260
+ ],
261
+ [
262
+ "Monotone vs. Expressive",
263
+ 0.0,
264
+ 4.0
265
+ ],
266
+ [
267
+ "High-Pitched vs. Low-Pitched",
268
+ 0.0,
269
+ 4.0
270
+ ],
271
+ [
272
+ "Soft vs. Harsh",
273
+ -2.0,
274
+ 2.0
275
+ ],
276
+ [
277
+ "Authenticity",
278
+ 0.0,
279
+ 4.0
280
+ ],
281
+ [
282
+ "Recording Quality",
283
+ 0.0,
284
+ 4.0
285
+ ],
286
+ [
287
+ "Background Noise",
288
+ 0.0,
289
+ 3.0
290
+ ]
291
+ ],
292
  "hidden_dropout_prob": 0.1,
293
+ "model_type": "bert-emotion-ordinal-regression",
 
 
 
 
 
 
 
 
 
294
  "torch_dtype": "float32",
295
+ "transformers_version": "4.49.0"
 
 
 
296
  }