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