metadata
license: mit
language:
- en
library_name: transformers
pipeline_tag: text-classification
tags:
- bert-base-uncased
- disease
- medical
widget:
- text: I am having itching, skin rash, and nodal skin eruptions
example_title: Fungal infection example
- text: I feel like vomiting, breathlessness, and sweating
example_title: Heart Attack example
- text: I am feeling fatigue, weight loss, restlessness and also lethargy.
example_title: Diabetes example
The objective is to develop a symptom-to-disease classification model for a natural language chatbot.
This model takes input text such as "I am feeling vomiting, breathlessness, and sweating" and accurately identifies the associated disease (Label_18 - 'Heart attack').
In essence, the chatbot's purpose is to analyze users' symptoms and provide relevant disease predictions in real-time conversation.
Superset of https://huggingface.co/shanover/disease_classifier_base
Symptoms to Disease - Sequence Classifier (41 Diseases)
List of Encoded Numbers and Corresponding Disease Names
- (Vertigo) Paroxysmal Positional Vertigo
- AIDS
- Acne
- Alcoholic hepatitis
- Allergy
- Arthritis
- Bronchial Asthma
- Cervical spondylosis
- Chicken pox
- Chronic cholestasis
- Common Cold
- Dengue
- Diabetes
- Dimorphic hemorrhoids (piles)
- Drug Reaction
- Fungal infection
- GERD
- Gastroenteritis
- Heart attack
- Hepatitis B
- Hepatitis C
- Hepatitis D
- Hepatitis E
- Hypertension
- Hyperthyroidism
- Hypoglycemia
- Hypothyroidism
- Impetigo
- Jaundice
- Malaria
- Migraine
- Osteoarthritis
- Paralysis (brain hemorrhage)
- Peptic ulcer disease
- Pneumonia
- Psoriasis
- Tuberculosis
- Typhoid
- Urinary tract infection
- Varicose veins
- Hepatitis A
EOD