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 (2 - 'Heart attack').
In essence, the chatbot's purpose is to analyze users' symptoms and provide relevant disease predictions in real-time conversation.
Labels:
0 - Fungal infection
1 - Diabetes
2 - Heart attack
Will add more diseases in coming days