Text Classification
Transformers
PyTorch
Safetensors
English
bert
bert-base-uncased
disease
medical
text-embeddings-inference
Instructions to use shanover/symps_disease_bert_v3_c41 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use shanover/symps_disease_bert_v3_c41 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="shanover/symps_disease_bert_v3_c41")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("shanover/symps_disease_bert_v3_c41") model = AutoModelForSequenceClassification.from_pretrained("shanover/symps_disease_bert_v3_c41") - Notebooks
- Google Colab
- Kaggle
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
- Downloads last month
- 414