Downloads · 30 days
26
4% of all-time downloads
disease_classifier_base
shanover/disease_classifier_base
disease_classifier_base is a text classification model from shanover. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
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…
All-time downloads
595
Public
Parameters
109M
876 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Jul 20, 2023
- Updated
- Jul 20, 2023
- SHA
- cf87b4ad
Try a prompt
Model card
From the Hugging Face model README
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
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- mit
- Languages
- en
- Created
- Jul 20, 2023
- Updated
- Jul 20, 2023