AyoubChLin
commited on
Commit
•
6c94387
1
Parent(s):
15e2204
Update config.json
Browse files- config.json +12 -12
config.json
CHANGED
@@ -28,22 +28,22 @@
|
|
28 |
"forced_eos_token_id": 2,
|
29 |
"gradient_checkpointing": false,
|
30 |
"id2label": {
|
31 |
-
"0": "
|
32 |
-
"1": "
|
33 |
-
"2": "
|
34 |
-
"3": "
|
35 |
-
"4": "
|
36 |
-
"5": "
|
37 |
},
|
38 |
"init_std": 0.02,
|
39 |
"is_encoder_decoder": true,
|
40 |
"label2id": {
|
41 |
-
"
|
42 |
-
"
|
43 |
-
"
|
44 |
-
"
|
45 |
-
"
|
46 |
-
"
|
47 |
},
|
48 |
"max_position_embeddings": 1024,
|
49 |
"model_type": "bart",
|
|
|
28 |
"forced_eos_token_id": 2,
|
29 |
"gradient_checkpointing": false,
|
30 |
"id2label": {
|
31 |
+
"0": "business",
|
32 |
+
"1": "entertainment",
|
33 |
+
"2": "health",
|
34 |
+
"3": "news",
|
35 |
+
"4": "politics",
|
36 |
+
"5": "sport"
|
37 |
},
|
38 |
"init_std": 0.02,
|
39 |
"is_encoder_decoder": true,
|
40 |
"label2id": {
|
41 |
+
"business": 0,
|
42 |
+
"entertainment": 1,
|
43 |
+
"health": 2,
|
44 |
+
"news": 3,
|
45 |
+
"politics": 4,
|
46 |
+
"sport": 5
|
47 |
},
|
48 |
"max_position_embeddings": 1024,
|
49 |
"model_type": "bart",
|