Instructions to use Charangan/MedBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Charangan/MedBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="Charangan/MedBERT")# Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("Charangan/MedBERT") model = AutoModelForPreTraining.from_pretrained("Charangan/MedBERT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ tags:
|
|
| 9 |
|
| 10 |
# MedBERT Model
|
| 11 |
|
| 12 |
-
MedBERT is a newly pre-trained transformer-based language model for biomedical named entity recognition: initialized with [Bio_ClinicalBERT](https://arxiv.org/abs/1904.03323) & pre-trained on N2C2, BioNLP, and CRAFT community datasets.
|
| 13 |
|
| 14 |
## Pretraining
|
| 15 |
|
|
@@ -28,7 +28,7 @@ The `MedBERT` model was trained on N2C2, BioNLP, and CRAFT community datasets.
|
|
| 28 |
The model was trained using code from [Google's BERT repository](https://github.com/google-research/bert). Model parameters were initialized with Bio_ClinicalBERT.
|
| 29 |
|
| 30 |
### Hyperparameters
|
| 31 |
-
We used a batch size of 32, a maximum sequence length of
|
| 32 |
and max predictions per sequence = 22).
|
| 33 |
|
| 34 |
## How to use
|
|
|
|
| 9 |
|
| 10 |
# MedBERT Model
|
| 11 |
|
| 12 |
+
**MedBERT** is a newly pre-trained transformer-based language model for biomedical named entity recognition: initialized with [Bio_ClinicalBERT](https://arxiv.org/abs/1904.03323) & pre-trained on N2C2, BioNLP, and CRAFT community datasets.
|
| 13 |
|
| 14 |
## Pretraining
|
| 15 |
|
|
|
|
| 28 |
The model was trained using code from [Google's BERT repository](https://github.com/google-research/bert). Model parameters were initialized with Bio_ClinicalBERT.
|
| 29 |
|
| 30 |
### Hyperparameters
|
| 31 |
+
We used a batch size of 32, a maximum sequence length of 256, and a learning rate of 1·10−4 for pre-training our models. The models trained for 200,000 steps. The dup factor for duplicating input data with different masks was set to 5. All other default parameters were used (specifically, masked language model probability = 0.15
|
| 32 |
and max predictions per sequence = 22).
|
| 33 |
|
| 34 |
## How to use
|