Downloads · 30 days
28
1% of all-time downloads
bigbird-base-mimic-mortality
mnaylor/bigbird-base-mimic-mortality
bigbird-base-mimic-mortality is a text classification model from mnaylor. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as bigscience-openrail-m.
Starting with Google's base BigBird model, we fine-tuned on binary mortality prediction in MIMIC admission notes. This model seeks to predict whether a certain patient will expire within a given ICU stay, based on the…
All-time downloads
4.1K
Public
Parameters
128M
1 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin512 MB · 50%
Parameter types
How the weights are stored.
F32128M · 100%
Try a prompt
Model card
From the Hugging Face model README
BigBird for Mortality Prediction
Starting with Google's base BigBird model, we fine-tuned on binary mortality prediction in MIMIC admission notes. This model seeks to predict whether a certain patient will expire within a given ICU stay, based on the text available upon admission. Data prepared for this task as described in this project, using the simulated admission notes (taken from discharge summaries). This model will be used in an upcoming submission for IMLH at ICML 2021.
References
- Van Aken, et al., 2021: Clinical Outcome Prediction from Admission Notes using Self-Supervised Knowledge Integration
- Zaheer, et al., 2020: Big Bird: Transformers for Longer Sequences
- Task
- Text Classification
- Library
- transformers
- Type
- big_bird
- License
- bigscience-openrail-m
- Created
- Mar 2, 2022
- Updated
- Jul 17, 2023