Update README.md

README.md

@@ -37,6 +37,12 @@ This model is intended for educational and demonstrational purposes only. It tak
 
 To use this model, you must load the feature extractor (`Bio_ClinicalBERT`), the LightGBM classifier, and the label encoder.
 
+## Training Data
+This model was trained on the gretelai/symptom_to_diagnosis dataset, which contains ~1000 symptom descriptions across 22 balanced classes.
+
+## Evaluation
+The model achieves a Macro F1-score of 0.834 and an Accuracy of 0.835 on the test set.
+
 ```python
 import torch
 import joblib