Update README.md
Browse files
README.md
CHANGED
|
@@ -21,6 +21,8 @@ model-index:
|
|
| 21 |
metrics:
|
| 22 |
- type: loss
|
| 23 |
value: 0.2577
|
|
|
|
|
|
|
| 24 |
---
|
| 25 |
|
| 26 |
# 🏥 SymbiPredict: ClinicalBERT Symptom-to-Disease Classifier
|
|
@@ -31,12 +33,11 @@ It has been trained on a massive merged dataset of over **96,000 patient cases**
|
|
| 31 |
|
| 32 |
## 📊 Model Performance
|
| 33 |
|
| 34 |
-
| Epoch | Training Loss | Validation Loss |
|
| 35 |
-
|-------|---------------|-----------------|
|
| 36 |
-
| 1 | 0.4108 | 0.3452 |
|
| 37 |
-
| 2 | 0.3092 | 0.2852 |
|
| 38 |
-
| 3 | 0.2526 | **0.2577**
|
| 39 |
-
|
| 40 |
The model achieves a final validation loss of **0.2577**, demonstrating high confidence and generalization capabilities across 115 disease classes.
|
| 41 |
|
| 42 |
## 🚀 How to Use (Python)
|
|
|
|
| 21 |
metrics:
|
| 22 |
- type: loss
|
| 23 |
value: 0.2577
|
| 24 |
+
base_model:
|
| 25 |
+
- emilyalsentzer/Bio_ClinicalBERT
|
| 26 |
---
|
| 27 |
|
| 28 |
# 🏥 SymbiPredict: ClinicalBERT Symptom-to-Disease Classifier
|
|
|
|
| 33 |
|
| 34 |
## 📊 Model Performance
|
| 35 |
|
| 36 |
+
| Epoch | Training Loss | Validation Loss |
|
| 37 |
+
|-------|---------------|-----------------|
|
| 38 |
+
| 1 | 0.4108 | 0.3452 |
|
| 39 |
+
| 2 | 0.3092 | 0.2852 |
|
| 40 |
+
| 3 | 0.2526 | **0.2577** |
|
|
|
|
| 41 |
The model achieves a final validation loss of **0.2577**, demonstrating high confidence and generalization capabilities across 115 disease classes.
|
| 42 |
|
| 43 |
## 🚀 How to Use (Python)
|