disease_classifier_base

作者 shanover · 下载 2 · 点赞 0 · 更新 2023-11-03

bert-base-uncased disease medical

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