Downloads · 30 days
17
1% of all-time downloads
mimic_roberta_base
xdai/mimic_roberta_base
mimic_roberta_base is a fill-mask model from xdai. Use it when you need the model to fill a missing word. It is set up for transformers. The card lists the license as cc-by-4.0.
Continue pre-training RoBERTa-base using discharge summaries from MIMIC-III datasets.
All-time downloads
1.4K
Public
Repo size
998 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin499 MB · 100%
At a glance
- Task
- Fill-Mask
- Library
- transformers
- License
- cc-by-4.0
- Model type
- roberta
- Access
- Public
- Created
- Jun 12, 2022
- Updated
- Jun 12, 2022
- SHA
- caa7fe3f
Try a prompt
Datasets
Model card
From the Hugging Face model README
-
Continue pre-training RoBERTa-base using discharge summaries from MIMIC-III datasets.
-
Details can be found in the following paper
Xiang Dai and Ilias Chalkidis and Sune Darkner and Desmond Elliott. 2022. Revisiting Transformer-based Models for Long Document Classification. (https://arxiv.org/abs/2204.06683)
- Important hyper-parameters
| Max sequence | 128 |
| Batch size | 128 |
| Learning rate | 5e-5 |
| Training epochs | 15 |
| Training time | 40 GPU-hours |
- Task
- Fill-Mask
- Library
- transformers
- Type
- roberta
- License
- cc-by-4.0
- Created
- Jun 12, 2022
- Updated
- Jun 12, 2022