AyoubChLin commited on
Commit
6c94387
1 Parent(s): 15e2204

Update config.json

Browse files
Files changed (1) hide show
  1. 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": "LABEL_0",
32
- "1": "LABEL_1",
33
- "2": "LABEL_2",
34
- "3": "LABEL_3",
35
- "4": "LABEL_4",
36
- "5": "LABEL_5"
37
  },
38
  "init_std": 0.02,
39
  "is_encoder_decoder": true,
40
  "label2id": {
41
- "LABEL_0": 0,
42
- "LABEL_1": 1,
43
- "LABEL_2": 2,
44
- "LABEL_3": 3,
45
- "LABEL_4": 4,
46
- "LABEL_5": 5
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",