https://github.com/scai-bio/exmed-bert
https://github.com/scai-bio/exmed-bert
Last synced: 23 days ago
JSON representation
- Host: GitHub
- URL: https://github.com/scai-bio/exmed-bert
- Owner: SCAI-BIO
- Created: 2022-11-15T09:03:24.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2025-02-03T15:23:24.000Z (4 months ago)
- Last Synced: 2025-04-30T06:07:32.832Z (23 days ago)
- Language: Python
- Size: 2.67 MB
- Stars: 6
- Watchers: 4
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# ExMed-BERT: A Transformer-Based Model Trained on Large Scale Claims Data for Prediction of Severe COVID Disease Progression
Code for ExMed-BERT training and fine-tuning.
## Pre-trained model
Our pre-trained model is available at https://doi.org/10.5281/zenodo.7324178.
## Relevant files
- **scripts/calculate\_iptw\_scores.py**: Script to calculate IPTWs
- **scripts/pretrain-exmed-bert.py**: Script to pre-train ExMed-BERT from scratch
- **scripts/finetune-exmed-bert.py**: Script to fine-tune ExMed-BERT for a classification task
- **scripts/train-rf.py**: Script to train an RF model for a classification task
- **scripts/train-xgboost.py**: Script to train an XGBoost model for a classification task
## Contact
Please post a GitHub issue or write an e-mail to [email protected] if you have any questions.