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