Gkumi commited on
Commit
fa6a019
·
verified ·
1 Parent(s): 0f362e7

Upload DistilBertForTokenClassification

Browse files
Files changed (2) hide show
  1. config.json +41 -41
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "distilbert-base-cased",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForTokenClassification"
@@ -10,49 +10,49 @@
10
  "hidden_dim": 3072,
11
  "id2label": {
12
  "0": "O",
13
- "1": "B-ACCOUNTNUMBER",
14
- "2": "I-ACCOUNTNUMBER",
15
- "3": "B-ADDRESS",
16
- "4": "I-ADDRESS",
17
- "5": "B-AGE",
18
- "6": "I-AGE",
19
- "7": "B-AMOUNT",
20
- "8": "I-AMOUNT",
21
- "9": "B-BIC",
22
- "10": "I-BIC",
23
- "11": "B-CITY",
24
- "12": "I-CITY",
25
- "13": "B-COMPANYNAME",
26
- "14": "I-COMPANYNAME",
27
- "15": "B-COUNTY",
28
- "16": "I-COUNTY",
29
- "17": "B-CREDITCARDCVV",
30
- "18": "I-CREDITCARDCVV",
31
- "19": "B-CREDITCARDNUMBER",
32
- "20": "I-CREDITCARDNUMBER"
33
  },
34
  "initializer_range": 0.02,
35
  "label2id": {
36
- "B-ACCOUNTNUMBER": 1,
37
- "B-ADDRESS": 3,
38
- "B-AGE": 5,
39
- "B-AMOUNT": 7,
40
- "B-BIC": 9,
41
- "B-CITY": 11,
42
- "B-COMPANYNAME": 13,
43
- "B-COUNTY": 15,
44
- "B-CREDITCARDCVV": 17,
45
- "B-CREDITCARDNUMBER": 19,
46
- "I-ACCOUNTNUMBER": 2,
47
- "I-ADDRESS": 4,
48
- "I-AGE": 6,
49
- "I-AMOUNT": 8,
50
- "I-BIC": 10,
51
- "I-CITY": 12,
52
- "I-COMPANYNAME": 14,
53
- "I-COUNTY": 16,
54
- "I-CREDITCARDCVV": 18,
55
- "I-CREDITCARDNUMBER": 20,
56
  "O": 0
57
  },
58
  "max_position_embeddings": 512,
 
1
  {
2
+ "_name_or_path": "Gkumi/low-level-training",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForTokenClassification"
 
10
  "hidden_dim": 3072,
11
  "id2label": {
12
  "0": "O",
13
+ "1": "B-DATE",
14
+ "2": "I-DATE",
15
+ "3": "B-DOB",
16
+ "4": "I-DOB",
17
+ "5": "B-EMAIL",
18
+ "6": "I-EMAIL",
19
+ "7": "B-GENDER",
20
+ "8": "I-GENDER",
21
+ "9": "B-HEIGHT",
22
+ "10": "I-HEIGHT",
23
+ "11": "B-IBAN",
24
+ "12": "I-IBAN",
25
+ "13": "B-IP",
26
+ "14": "I-IP",
27
+ "15": "B-IPV4",
28
+ "16": "I-IPV4",
29
+ "17": "B-IPV6",
30
+ "18": "I-IPV6",
31
+ "19": "B-JOBTITLE",
32
+ "20": "I-JOBTITLE"
33
  },
34
  "initializer_range": 0.02,
35
  "label2id": {
36
+ "B-DATE": 1,
37
+ "B-DOB": 3,
38
+ "B-EMAIL": 5,
39
+ "B-GENDER": 7,
40
+ "B-HEIGHT": 9,
41
+ "B-IBAN": 11,
42
+ "B-IP": 13,
43
+ "B-IPV4": 15,
44
+ "B-IPV6": 17,
45
+ "B-JOBTITLE": 19,
46
+ "I-DATE": 2,
47
+ "I-DOB": 4,
48
+ "I-EMAIL": 6,
49
+ "I-GENDER": 8,
50
+ "I-HEIGHT": 10,
51
+ "I-IBAN": 12,
52
+ "I-IP": 14,
53
+ "I-IPV4": 16,
54
+ "I-IPV6": 18,
55
+ "I-JOBTITLE": 20,
56
  "O": 0
57
  },
58
  "max_position_embeddings": 512,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8cf4aa9973208671a715c01b6e0c729f8b48423e0bf9f915b926321d1b1845e0
3
  size 260840580
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5de6583a4a6d9b01b0a4fee2657adfdbe066822f6e3e520cf30995d38f1155b
3
  size 260840580