HiAmNear commited on
Commit
03731e5
·
verified ·
1 Parent(s): 9b8c222

Upload RobertaForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +22 -12
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "_name_or_path": "../../modelParams/phobert-base-ViFE",
3
  "architectures": [
4
- "RobertaModel"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bos_token_id": 0,
@@ -12,19 +12,29 @@
12
  "hidden_dropout_prob": 0.1,
13
  "hidden_size": 768,
14
  "id2label": {
15
- "0": "HAPPINESS",
16
- "1": "SADNESS",
17
- "2": "NEUTRAL",
18
- "3": "ANGER",
19
- "4": "AFFECTION",
20
- "5": "BRAVERY",
21
- "6": "FEAR",
22
- "7": "DISGUST",
23
- "8": "SURPRISE"
24
  },
25
  "initializer_range": 0.02,
26
  "intermediate_size": 3072,
27
- "label2id": null,
 
 
 
 
 
 
 
 
 
 
28
  "layer_norm_eps": 1e-05,
29
  "max_position_embeddings": 258,
30
  "model_type": "roberta",
 
1
  {
2
+ "_name_or_path": "HiAmNear/phobert-base-ViFE",
3
  "architectures": [
4
+ "RobertaForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bos_token_id": 0,
 
12
  "hidden_dropout_prob": 0.1,
13
  "hidden_size": 768,
14
  "id2label": {
15
+ "0": "0",
16
+ "1": "1",
17
+ "2": "2",
18
+ "3": "3",
19
+ "4": "4",
20
+ "5": "5",
21
+ "6": "6",
22
+ "7": "7",
23
+ "8": "8"
24
  },
25
  "initializer_range": 0.02,
26
  "intermediate_size": 3072,
27
+ "label2id": {
28
+ "0": 0,
29
+ "1": 1,
30
+ "2": 2,
31
+ "3": 3,
32
+ "4": 4,
33
+ "5": 5,
34
+ "6": 6,
35
+ "7": 7,
36
+ "8": 8
37
+ },
38
  "layer_norm_eps": 1e-05,
39
  "max_position_embeddings": 258,
40
  "model_type": "roberta",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b0b112c99297fd35149c5409fa6106ae0efb28af19e8bde544c34ff002653ace
3
- size 540015464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52a6e013c22b5c2adf5df68d65e77dd9488a2a674fc80dd9742d4a9d393bd0ba
3
+ size 540044916