Version 2
Browse files- config.json +4 -4
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "distilbert-base-uncased",
|
3 |
"activation": "gelu",
|
4 |
"architectures": [
|
5 |
"DistilBertForSequenceClassification"
|
@@ -18,11 +18,11 @@
|
|
18 |
},
|
19 |
"initializer_range": 0.02,
|
20 |
"label2id": {
|
21 |
-
"sadness": 0,
|
22 |
-
"joy": 1,
|
23 |
-
"love": 2,
|
24 |
"anger": 3,
|
25 |
"fear": 4,
|
|
|
|
|
|
|
26 |
"surprise": 5
|
27 |
},
|
28 |
"max_position_embeddings": 512,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "reboot87/distilbert-base-uncased-finetuned-emotions",
|
3 |
"activation": "gelu",
|
4 |
"architectures": [
|
5 |
"DistilBertForSequenceClassification"
|
|
|
18 |
},
|
19 |
"initializer_range": 0.02,
|
20 |
"label2id": {
|
|
|
|
|
|
|
21 |
"anger": 3,
|
22 |
"fear": 4,
|
23 |
+
"joy": 1,
|
24 |
+
"love": 2,
|
25 |
+
"sadness": 0,
|
26 |
"surprise": 5
|
27 |
},
|
28 |
"max_position_embeddings": 512,
|