Downloads · 30 days
0
arshy/medicalspecialty-deberta
medicalspecialty-deberta is a machine learning model from arshy. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for fastai.
Downloads · 30 days
0
Access
Public
Updated Jun 16, 2022
Repo size
—
Likes
1
Public
Click a slice to open those files.
.pkl1.1 GB · 100%
From the Hugging Face model README
#Project Description
The project aims to classify a given text input of symptoms into a medical specialty that might be relevant to the symptoms. The model classifies the input into one of the 28 specialties, such as gastroenterologist, pediatrics, etc., that it was trained in.
#Model Description
This model is a fine-tuned version of deberta-v3-small using the amazing Blurr library (huggingface wrapper library using fastai)
#Dataset Used
The dataset for this task was built by scrapping Patient.info. In the forum, posts are organized under different categories/groups such as anxiety disorder, acne, malaria, etc. These categories/groups were manually mapped to a specialty for eg anxiety to a psychiatrist, and malaria to an infectious disease specialist.
##Others
I have a blog on the same task using fastai's ULMFiT model. That can be found here.
Disclaimer This project is a fun project intended for learning purposes only.