connorboyle commited on
Commit
6afcc8b
1 Parent(s): fdcf523

Add Initial Model + README

Browse files
I2B2-NER-5-Epochs-Model/config.json ADDED
@@ -0,0 +1,124 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-cased",
3
+ "architectures": [
4
+ "BertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": [
13
+ "O",
14
+ "B-AGE",
15
+ "I-AGE",
16
+ "B-BIOID",
17
+ "I-BIOID",
18
+ "B-CITY",
19
+ "I-CITY",
20
+ "B-COUNTRY",
21
+ "I-COUNTRY",
22
+ "B-DATE",
23
+ "I-DATE",
24
+ "B-DEVICE",
25
+ "I-DEVICE",
26
+ "B-DOCTOR",
27
+ "I-DOCTOR",
28
+ "B-EMAIL",
29
+ "I-EMAIL",
30
+ "B-FAX",
31
+ "I-FAX",
32
+ "B-HEALTHPLAN",
33
+ "I-HEALTHPLAN",
34
+ "B-HOSPITAL",
35
+ "I-HOSPITAL",
36
+ "B-IDNUM",
37
+ "I-IDNUM",
38
+ "B-LOCATION-OTHER",
39
+ "I-LOCATION-OTHER",
40
+ "B-MEDICALRECORD",
41
+ "I-MEDICALRECORD",
42
+ "B-ORGANIZATION",
43
+ "I-ORGANIZATION",
44
+ "B-PATIENT",
45
+ "I-PATIENT",
46
+ "B-PHONE",
47
+ "I-PHONE",
48
+ "B-PROFESSION",
49
+ "I-PROFESSION",
50
+ "B-STATE",
51
+ "I-STATE",
52
+ "B-STREET",
53
+ "I-STREET",
54
+ "B-URL",
55
+ "I-URL",
56
+ "B-USERNAME",
57
+ "I-USERNAME",
58
+ "B-ZIP",
59
+ "I-ZIP"
60
+ ],
61
+ "initializer_range": 0.02,
62
+ "intermediate_size": 3072,
63
+ "label2id": {
64
+ "B-AGE": 1,
65
+ "B-BIOID": 3,
66
+ "B-CITY": 5,
67
+ "B-COUNTRY": 7,
68
+ "B-DATE": 9,
69
+ "B-DEVICE": 11,
70
+ "B-DOCTOR": 13,
71
+ "B-EMAIL": 15,
72
+ "B-FAX": 17,
73
+ "B-HEALTHPLAN": 19,
74
+ "B-HOSPITAL": 21,
75
+ "B-IDNUM": 23,
76
+ "B-LOCATION-OTHER": 25,
77
+ "B-MEDICALRECORD": 27,
78
+ "B-ORGANIZATION": 29,
79
+ "B-PATIENT": 31,
80
+ "B-PHONE": 33,
81
+ "B-PROFESSION": 35,
82
+ "B-STATE": 37,
83
+ "B-STREET": 39,
84
+ "B-URL": 41,
85
+ "B-USERNAME": 43,
86
+ "B-ZIP": 45,
87
+ "I-AGE": 2,
88
+ "I-BIOID": 4,
89
+ "I-CITY": 6,
90
+ "I-COUNTRY": 8,
91
+ "I-DATE": 10,
92
+ "I-DEVICE": 12,
93
+ "I-DOCTOR": 14,
94
+ "I-EMAIL": 16,
95
+ "I-FAX": 18,
96
+ "I-HEALTHPLAN": 20,
97
+ "I-HOSPITAL": 22,
98
+ "I-IDNUM": 24,
99
+ "I-LOCATION-OTHER": 26,
100
+ "I-MEDICALRECORD": 28,
101
+ "I-ORGANIZATION": 30,
102
+ "I-PATIENT": 32,
103
+ "I-PHONE": 34,
104
+ "I-PROFESSION": 36,
105
+ "I-STATE": 38,
106
+ "I-STREET": 40,
107
+ "I-URL": 42,
108
+ "I-USERNAME": 44,
109
+ "I-ZIP": 46,
110
+ "O": 0
111
+ },
112
+ "layer_norm_eps": 1e-12,
113
+ "max_position_embeddings": 512,
114
+ "model_type": "bert",
115
+ "num_attention_heads": 12,
116
+ "num_hidden_layers": 12,
117
+ "pad_token_id": 0,
118
+ "position_embedding_type": "absolute",
119
+ "torch_dtype": "float32",
120
+ "transformers_version": "4.10.0",
121
+ "type_vocab_size": 2,
122
+ "use_cache": true,
123
+ "vocab_size": 28996
124
+ }
I2B2-NER-5-Epochs-Model/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:014029d9f4a39272699d8cc5a35c9ba1c9d6899c2a75d5e305bf9dc5a6642e6d
3
+ size 431103857
README.md ADDED
@@ -0,0 +1 @@
 
 
1
+ Named-entity recognition model trained on the I2B2 training data set for PHI.