---
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

0. (Vertigo) Paroxysmal Positional Vertigo
1. AIDS
2. Acne
3. Alcoholic hepatitis
4. Allergy
5. Arthritis
6. Bronchial Asthma
7. Cervical spondylosis
8. Chicken pox
9. Chronic cholestasis
10. Common Cold
11. Dengue
12. Diabetes
13. Dimorphic hemorrhoids (piles)
14. Drug Reaction
15. Fungal infection
16. GERD
17. Gastroenteritis
18. Heart attack
19. Hepatitis B
20. Hepatitis C
21. Hepatitis D
22. Hepatitis E
23. Hypertension
24. Hyperthyroidism
25. Hypoglycemia
26. Hypothyroidism
27. Impetigo
28. Jaundice
29. Malaria
30. Migraine
31. Osteoarthritis
32. Paralysis (brain hemorrhage)
33. Peptic ulcer disease
34. Pneumonia
35. Psoriasis
36. Tuberculosis
37. Typhoid
38. Urinary tract infection
39. Varicose veins
40. Hepatitis A

EOD