Create README.md
Browse files# AI-Powered Symptom Checker 🏥🤖
This model predicts potential medical conditions based on user-reported symptoms. Built using **BERT** and fine-tuned on the **MedText dataset**, it helps users get preliminary symptom insights.
## 🔍 Model Details
- **Model Type:** Text Classification
- **Base Model:** BERT (`bert-base-uncased`)
- **Dataset:** MedText (1.4k medical cases)
- **Metrics:** Accuracy: `96.5%`, F1-score: `95.1%`
- **Intended Use:** Assist users in identifying possible conditions based on symptoms
- **Limitations:** Not a replacement for professional medical diagnosis
## 📖 Usage Example
```python
from transformers import pipeline
model = pipeline("text-classification", model="Lech-Iyoko/bert-symptom-checker")
result = model("I have a severe headache and nausea.")
print(result)
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
datasets:
|
4 |
+
- BI55/MedText
|
5 |
+
language:
|
6 |
+
- en
|
7 |
+
metrics:
|
8 |
+
- accuracy
|
9 |
+
- f1
|
10 |
+
base_model:
|
11 |
+
- google-bert/bert-base-uncased
|
12 |
+
pipeline_tag: text-classification
|
13 |
+
library_name: transformers
|
14 |
+
tags:
|
15 |
+
- medical
|
16 |
+
- symptomchecker
|
17 |
+
- nlp
|
18 |
+
- healthcare
|
19 |
+
---
|