lxhlxhlxh555
commited on
Commit
•
c3a9469
1
Parent(s):
4ead442
Update README.md
Browse files
README.md
CHANGED
@@ -7,7 +7,7 @@ tags:
|
|
7 |
<!-- Provide a quick summary of what the model is/does. -->
|
8 |
|
9 |
This model card describes the ClinicalBERT model, which was trained on a large multicenter dataset with a large corpus of 1.2B words of diverse diseases we constructed.
|
10 |
-
We then utilized a
|
11 |
|
12 |
## Pretraining Data
|
13 |
|
@@ -19,14 +19,11 @@ For more details, see here.
|
|
19 |
### Pretraining Procedures
|
20 |
The ClinicalBERT was initialized from BERT. Then the training followed the principle of masked language model, in which given a piece of text, we randomly replace some tokens by MASKs,
|
21 |
special tokens for masking, and then require the model to predict the original tokens via contextual text.
|
22 |
-
The training code can be found [here](https://www.github.com/xxx) and the model was trained on four A100 GPU.
|
23 |
|
24 |
### Pretraining Hyperparameters
|
25 |
|
26 |
We used a batch size of xx, a maximum sequence length of xx, and a learning rate of xx for pre-training our models.
|
27 |
The model was trained for xx steps.
|
28 |
-
The dup factor for duplicating input data with different masks was set to 5.
|
29 |
-
All other default parameters were used (xxx).
|
30 |
|
31 |
## How to use the model
|
32 |
|
@@ -35,13 +32,4 @@ Load the model via the transformers library:
|
|
35 |
from transformers import AutoTokenizer, AutoModel
|
36 |
tokenizer = AutoTokenizer.from_pretrained("medicalai/ClinicalBERT")
|
37 |
model = AutoModel.from_pretrained("medicalai/ClinicalBERT")
|
38 |
-
|
39 |
```
|
40 |
-
|
41 |
-
## More Information
|
42 |
-
|
43 |
-
Refer to the paper xxx.
|
44 |
-
|
45 |
-
## Questions?
|
46 |
-
|
47 |
-
Post a Github issue on the xxx repo or email xxx with any questions.
|
|
|
7 |
<!-- Provide a quick summary of what the model is/does. -->
|
8 |
|
9 |
This model card describes the ClinicalBERT model, which was trained on a large multicenter dataset with a large corpus of 1.2B words of diverse diseases we constructed.
|
10 |
+
We then utilized a large-scale corpus of EHRs from over 3 million pediatric outpatient records to fine tune the base language model.
|
11 |
|
12 |
## Pretraining Data
|
13 |
|
|
|
19 |
### Pretraining Procedures
|
20 |
The ClinicalBERT was initialized from BERT. Then the training followed the principle of masked language model, in which given a piece of text, we randomly replace some tokens by MASKs,
|
21 |
special tokens for masking, and then require the model to predict the original tokens via contextual text.
|
|
|
22 |
|
23 |
### Pretraining Hyperparameters
|
24 |
|
25 |
We used a batch size of xx, a maximum sequence length of xx, and a learning rate of xx for pre-training our models.
|
26 |
The model was trained for xx steps.
|
|
|
|
|
27 |
|
28 |
## How to use the model
|
29 |
|
|
|
32 |
from transformers import AutoTokenizer, AutoModel
|
33 |
tokenizer = AutoTokenizer.from_pretrained("medicalai/ClinicalBERT")
|
34 |
model = AutoModel.from_pretrained("medicalai/ClinicalBERT")
|
|
|
35 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|