Reboot87 commited on
Commit
782e04f
1 Parent(s): d61236b
Files changed (1) hide show
  1. 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,