Hugging Face's logo

AyoubChLin
/
Albert-bbc-news

Text Classification
Transformers
PyTorch
Safetensors
English
albert
Trained with AutoTrain
Inference Endpoints
Model card Files Community
1
AyoubChLin commited on
Commit
cf7b388
·
1 Parent(s): e0c2900

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -11
config.json CHANGED
@@ -15,21 +15,21 @@
15
  "hidden_dropout_prob": 0,
16
  "hidden_size": 4096,
17
  "id2label": {
18
- "0": "0",
19
- "1": "1",
20
- "2": "2",
21
- "3": "3",
22
- "4": "4"
23
  },
24
  "initializer_range": 0.02,
25
  "inner_group_num": 1,
26
  "intermediate_size": 16384,
27
- "label2id": {
28
- "0": 0,
29
- "1": 1,
30
- "2": 2,
31
- "3": 3,
32
- "4": 4
33
  },
34
  "layer_norm_eps": 1e-12,
35
  "layers_to_keep": [],
 
15
  "hidden_dropout_prob": 0,
16
  "hidden_size": 4096,
17
  "id2label": {
18
+ "0": "tech",
19
+ "1": "business",
20
+ "2": "sport",
21
+ "3": "entertainment",
22
+ "4": "politics"
23
  },
24
  "initializer_range": 0.02,
25
  "inner_group_num": 1,
26
  "intermediate_size": 16384,
27
+ "id2label": {
28
+ "0": "tech",
29
+ "1": "business",
30
+ "2": "sport",
31
+ "3": "entertainment",
32
+ "4": "politics"
33
  },
34
  "layer_norm_eps": 1e-12,
35
  "layers_to_keep": [],