Downloads · 30 days
10
2% of all-time downloads
identifying_pregnancy_clinical_notes
elliealbertson/identifying_pregnancy_clinical_notes
identifying_pregnancy_clinical_notes is a text classification model from elliealbertson. Use it when you need a label for a piece of text. It is set up for transformers.
This model is a fine-tuned version of Bio+ClincialBert. It was fine-tuned on an open source Kaggle dataset of medical transcriptions with an aim of classifying clincial notes as discussing or not discussing a patient'…
All-time downloads
646
Public
Parameters
108M
867 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors433 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Dec 20, 2023
- Updated
- Dec 22, 2023
- SHA
- 20b97ffb
Try a prompt
Model card
From the Hugging Face model README
Model Card for identifying_pregnancy_clinical_notes
This model is a fine-tuned version of Bio+ClincialBert. It was fine-tuned on an open source Kaggle dataset of medical transcriptions with an aim of classifying clincial notes as discussing or not discussing a patient's pregnancy.
- Repository: Available on GitHub.
- Uses: Intended use is binary classification of English-language clinical notes as discussing or not discussing a patient's pregnancy.
- Limitations: The model was fine-tuned on a small number of clinical notes agumented by limited synthetic data. As a result, it may give inaccurate results at times. It is recommended to consider probability of class assignment when evaluating clinical note classifications.
How to Get Started with the Model
Load the model via the transformers library:
model_path = "elliealbertson/identifying_pregnancy_clinical_notes"
tokenizer = BertTokenizer.from_pretrained(model_path)
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Dec 20, 2023
- Updated
- Dec 22, 2023