Text Classification
Transformers
TensorFlow
English
distilbert
ditilbert
text classification
clinical notes
wellnation
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("oyesaurav/dwellbert")
model = AutoModelForSequenceClassification.from_pretrained("oyesaurav/dwellbert")Quick Links
labels map =
{
"0": "Gastroenterology",
"1": "Neurology",
"2": "Orthopedic",
"3": "Radiology",
"4": "Urology"
}
The fine tuned model has been trained on around 2300 medical transcriptions, to classify medical specialty. More classes will be added as data would be available.
- Downloads last month
- 17
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="oyesaurav/dwellbert")