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