AyoubChLin commited on
Commit
ec54430
1 Parent(s): f94d9d0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -14
config.json CHANGED
@@ -29,23 +29,23 @@
29
  "force_bos_token_to_be_generated": false,
30
  "forced_eos_token_id": 2,
31
  "gradient_checkpointing": false,
32
- "id2label": {
33
- "0": "LABEL_0",
34
- "1": "LABEL_1",
35
- "2": "LABEL_2",
36
- "3": "LABEL_3",
37
- "4": "LABEL_4",
38
- "5": "LABEL_5"
39
  },
40
  "init_std": 0.02,
41
  "is_encoder_decoder": true,
42
- "label2id": {
43
- "LABEL_0": 0,
44
- "LABEL_1": 1,
45
- "LABEL_2": 2,
46
- "LABEL_3": 3,
47
- "LABEL_4": 4,
48
- "LABEL_5": 5
49
  },
50
  "max_position_embeddings": 1024,
51
  "model_type": "bart",
 
29
  "force_bos_token_to_be_generated": false,
30
  "forced_eos_token_id": 2,
31
  "gradient_checkpointing": false,
32
+ "id2label": {
33
+ "0": "business",
34
+ "1": "entertainment",
35
+ "2": "health",
36
+ "3": "news",
37
+ "4": "politics",
38
+ "5": "sport"
39
  },
40
  "init_std": 0.02,
41
  "is_encoder_decoder": true,
42
+ "label2id": {
43
+ "business": 0,
44
+ "entertainment": 1,
45
+ "health": 2,
46
+ "news": 3,
47
+ "politics": 4,
48
+ "sport": 5
49
  },
50
  "max_position_embeddings": 1024,
51
  "model_type": "bart",