shanover's picture
Update README.md
a28985b
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

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

EOD