File size: 987 Bytes
bc48076
 
16df28f
 
 
 
 
 
 
 
62b60e1
 
 
 
 
 
 
bc48076
16df28f
62b60e1
cf87b4a
cf3770a
 
cf87b4a
 
 
 
cf3770a
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
---
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