Hugging Face's logo

Cheykong
/
CKwork

Text Classification
Transformers
Safetensors
distilbert
text-embeddings-inference
Model card Files
xet
 Community
Cheykong commited on
Commit
fc59f59
·
verified ·
1 Parent(s): 7adc74b

Upload DistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +12 -16
  2. model.safetensors +2 -2
config.json CHANGED
@@ -4,39 +4,35 @@
4
  "DistilBertForSequenceClassification"
5
  ],
6
  "attention_dropout": 0.1,
7
- "bos_token_id": null,
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "dtype": "float32",
11
- "eos_token_id": null,
12
  "hidden_dim": 3072,
13
  "id2label": {
14
- "0": "sadness",
15
- "1": "joy",
16
- "2": "love",
17
- "3": "anger",
18
- "4": "fear",
19
- "5": "surprise"
20
  },
21
  "initializer_range": 0.02,
22
  "label2id": {
23
- "anger": 3,
24
- "fear": 4,
25
- "joy": 1,
26
- "love": 2,
27
- "sadness": 0,
28
- "surprise": 5
29
  },
30
  "max_position_embeddings": 512,
31
  "model_type": "distilbert",
32
  "n_heads": 12,
33
  "n_layers": 6,
34
  "pad_token_id": 0,
 
35
  "qa_dropout": 0.1,
36
  "seq_classif_dropout": 0.2,
37
  "sinusoidal_pos_embds": false,
38
  "tie_weights_": true,
39
- "tie_word_embeddings": true,
40
- "transformers_version": "5.0.0",
41
  "vocab_size": 30522
42
  }
 
4
  "DistilBertForSequenceClassification"
5
  ],
6
  "attention_dropout": 0.1,
 
7
  "dim": 768,
8
  "dropout": 0.1,
9
  "dtype": "float32",
 
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3",
16
+ "4": "LABEL_4"
 
17
  },
18
  "initializer_range": 0.02,
19
  "label2id": {
20
+ "LABEL_0": 0,
21
+ "LABEL_1": 1,
22
+ "LABEL_2": 2,
23
+ "LABEL_3": 3,
24
+ "LABEL_4": 4
 
25
  },
26
  "max_position_embeddings": 512,
27
  "model_type": "distilbert",
28
  "n_heads": 12,
29
  "n_layers": 6,
30
  "pad_token_id": 0,
31
+ "problem_type": "single_label_classification",
32
  "qa_dropout": 0.1,
33
  "seq_classif_dropout": 0.2,
34
  "sinusoidal_pos_embds": false,
35
  "tie_weights_": true,
36
+ "transformers_version": "4.57.6",
 
37
  "vocab_size": 30522
38
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:229352dec2a292d7204f8873d35c6a5c34fd848f7b0e13e6d9f438324c4172de
3
- size 267844872
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f99263ccad6b67e8b7ff355925667194a50dab35a626c3d18f0db19d6ea9af0d
3
+ size 267841796