Hugging Face's logo

sandipan14
/
fine-grained

Text Classification
Safetensors
Model card Files
xet
 Community
sandipan14 commited on
Commit
7369dba
·
verified ·
1 Parent(s): 4004fac

Update fine_tuned/config.json

Browse files
Files changed (1) hide show
  1. fine_tuned/config.json +35 -34
fine_tuned/config.json CHANGED
@@ -1,34 +1,35 @@
1
- {
2
- "activation": "gelu",
3
- "architectures": [
4
- "DistilBertForMaskedLM"
5
- ],
6
- "attention_dropout": 0.1,
7
- "dim": 768,
8
- "dropout": 0.1,
9
- "hidden_dim": 3072,
10
- "id2label": {
11
- "0": "World",
12
- "1": "Sports",
13
- "2": "Business",
14
- "3": "Sci/Tech"
15
- },
16
- "initializer_range": 0.02,
17
- "label2id": {
18
- "Business": 2,
19
- "Sci/Tech": 3,
20
- "Sports": 1,
21
- "World": 0
22
- },
23
- "max_position_embeddings": 512,
24
- "model_type": "distilbert",
25
- "n_heads": 12,
26
- "n_layers": 6,
27
- "pad_token_id": 0,
28
- "qa_dropout": 0.1,
29
- "seq_classif_dropout": 0.2,
30
- "sinusoidal_pos_embds": false,
31
- "tie_weights_": true,
32
- "transformers_version": "4.57.1",
33
- "vocab_size": 30522
34
- }
 
 
1
+ {
2
+ "activation": "gelu",
3
+ "architectures": [
4
+ "DistilBertForSequenceClassification"
5
+ ],
6
+ "attention_dropout": 0.1,
7
+ "dim": 768,
8
+ "dropout": 0.1,
9
+ "hidden_dim": 3072,
10
+ "id2label": {
11
+ "0": "World",
12
+ "1": "Sports",
13
+ "2": "Business",
14
+ "3": "Sci/Tech"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "label2id": {
18
+ "World": 0,
19
+ "Sports": 1,
20
+ "Business": 2,
21
+ "Sci/Tech": 3
22
+ },
23
+ "max_position_embeddings": 512,
24
+ "model_type": "distilbert",
25
+ "n_heads": 12,
26
+ "n_layers": 6,
27
+ "pad_token_id": 0,
28
+ "qa_dropout": 0.1,
29
+ "seq_classif_dropout": 0.2,
30
+ "sinusoidal_pos_embds": false,
31
+ "tie_weights_": true,
32
+ "transformers_version": "4.57.1",
33
+ "vocab_size": 30522,
34
+ "num_labels": 4
35
+ }