CoolFace
Modelpublic

abhirajeshbhai/symptom-2-disease-net

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
2likes12downloads
Model Card

<!-- 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 LossTrain AccuracyValidation LossValidation AccuracyEpoch
2.66130.29541.72060.71670
1.13250.84910.66830.90001
0.40550.97500.28900.94172
0.15510.99720.19120.96673
0.08031.00.12960.95834

Framework versions

  • —Transformers 4.28.1
  • —TensorFlow 2.12.0
  • —Tokenizers 0.13.3