Edit model card

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

Downloads last month
105
Safetensors
Model size
110M params
Tensor type
F32
·

Space using shanover/symps_disease_bert_v3_c41 1