Downloads · 30 days
5
7% of all-time downloads
clinicalbert-finetuned
SrinivasaPragada/clinicalbert-finetuned
clinicalbert-finetuned is a text classification model from SrinivasaPragada. Use it when you need a label for a piece of text. It is set up for transformers.
This repository contains a fine-tuned version of the BioClinicalBERT model for classifying clinical text into various sections such as chief complaints, medications, and lab tests.
All-time downloads
67
Public
Repo size
3.5 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin433 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jul 1, 2024
- Updated
- Jul 1, 2024
- SHA
- 49eb53cc
Try a prompt
Model card
From the Hugging Face model README
ClinicalBERT Fine-Tuned Model
This repository contains a fine-tuned version of the Bio_ClinicalBERT model for classifying clinical text into various sections such as chief complaints, medications, and lab tests.
Model Details
- Base Model: Bio_ClinicalBERT
- Fine-Tuned On: Clinical text data
- Labels: chief complaints, medications, lab tests
Usage
You can use this model with the Hugging Face Transformers library:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("SrinivasaPragada/clinicalbert-finetuned")
model = AutoModelForSequenceClassification.from_pretrained("SrinivasaPragada/clinicalbert-finetuned")
inputs = tokenizer("Patient reports severe headaches and dizziness.", return_tensors="pt")
outputs = model(**inputs)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Jul 1, 2024
- Updated
- Jul 1, 2024