Hugging Face's logo

ma2za
/
roberta-emotion

Text Classification
Transformers
PyTorch
English
ma2za/roberta-emotion
feature-extraction
Emotion Classification
custom_code
Eval Results
Model card Files Community
1
ma2za commited on
Commit
ab19c17
·
1 Parent(s): 0cf1638

Upload RobertaForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +12 -12
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -11,22 +11,22 @@
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
  "id2label": {
14
- "0": "LABEL_0",
15
- "1": "LABEL_1",
16
- "2": "LABEL_2",
17
- "3": "LABEL_3",
18
- "4": "LABEL_4",
19
- "5": "LABEL_5"
20
  },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
- "LABEL_0": 0,
25
- "LABEL_1": 1,
26
- "LABEL_2": 2,
27
- "LABEL_3": 3,
28
- "LABEL_4": 4,
29
- "LABEL_5": 5
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "max_position_embeddings": 514,
 
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
  "id2label": {
14
+ "0": "sadness",
15
+ "1": "joy",
16
+ "2": "love",
17
+ "3": "anger",
18
+ "4": "fear",
19
+ "5": "surprise"
20
  },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
+ "anger": 3,
25
+ "fear": 4,
26
+ "joy": 1,
27
+ "love": 2,
28
+ "sadness": 0,
29
+ "surprise": 5
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "max_position_embeddings": 514,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b6e1d6a7292b9314e5be06281d82ccbd2de339acce80843994ed6687ba71ca8f
3
  size 498674357
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:407ae2a6b436ce83efdb7963c7fa5feb745955a7b1c5605ff37bbc3e9348d6bf
3
  size 498674357