Downloads · 30 days
6
0% of all-time downloads
symptom-2-disease-net
abhirajeshbhai/symptom-2-disease-net
symptom-2-disease-net is a text classification model from abhirajeshbhai. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
probably proofread and complete it, then remove this comment. --
All-time downloads
1.4K
Public
Repo size
3.9 GB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.h5434 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- Apr 27, 2023
- Updated
- Apr 27, 2023
- SHA
- 6589cefc
Try a prompt
Model card
From the Hugging Face model README
<!-- This model card has been generated automatically according to the information Keras had access to. You should
probably proofread and complete it, then remove this comment. -->
Model description
This model is trained on a Symptoms to Disease Dataset. The model takes in a symptom prompt and classifies the most probable disease from it.
Training results
| Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
|---|---|---|---|---|
| 2.6613 | 0.2954 | 1.7206 | 0.7167 | 0 |
| 1.1325 | 0.8491 | 0.6683 | 0.9000 | 1 |
| 0.4055 | 0.9750 | 0.2890 | 0.9417 | 2 |
| 0.1551 | 0.9972 | 0.1912 | 0.9667 | 3 |
| 0.0803 | 1.0 | 0.1296 | 0.9583 | 4 |
Framework versions
- Transformers 4.28.1
- TensorFlow 2.12.0
- Tokenizers 0.13.3
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Languages
- en
- Created
- Apr 27, 2023
- Updated
- Apr 27, 2023