Hugging Face's logo

Qishuai
/
distilbert_punctuator_en

Token Classification
Transformers
PyTorch
Safetensors
distilbert
Inference Endpoints
Model card Files Community
1
Qishuai commited on
Commit
6e13513
•
1 Parent(s): c1166ee

add model

Browse files
Files changed (2) hide show
  1. config.json +12 -12
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "models/punctuator",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForTokenClassification"
@@ -9,19 +9,19 @@
9
  "dropout": 0.2,
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",
@@ -33,6 +33,6 @@
33
  "sinusoidal_pos_embds": false,
34
  "tie_weights_": true,
35
  "torch_dtype": "float32",
36
- "transformers_version": "4.12.3",
37
  "vocab_size": 30522
38
  }
 
1
  {
2
+ "_name_or_path": "models/english_punctuator",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForTokenClassification"
 
9
  "dropout": 0.2,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "Q",
13
+ "1": "P",
14
+ "2": "O",
15
+ "3": "C",
16
+ "4": "E"
17
  },
18
  "initializer_range": 0.02,
19
  "label2id": {
20
+ "C": 3,
21
+ "E": 4,
22
+ "O": 2,
23
+ "P": 1,
24
+ "Q": 0
25
  },
26
  "max_position_embeddings": 512,
27
  "model_type": "distilbert",
 
33
  "sinusoidal_pos_embds": false,
34
  "tie_weights_": true,
35
  "torch_dtype": "float32",
36
+ "transformers_version": "4.12.5",
37
  "vocab_size": 30522
38
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cf088f8e114ee6ccc5ed924d26056d3c4788f77a57aa627d04362dbde1df0b62
3
  size 265508139
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc259d73be232fc0a2e1b7b681ca738bd0a7bda5028e3ba5446a19441ca20ca8
3
  size 265508139