Iloriayomide commited on
Commit
cc32efc
·
verified ·
1 Parent(s): 5de14dd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -1,11 +1,14 @@
1
  ---
2
  language: en
 
 
3
  tags:
4
  - medical
5
  - classification
6
  - healthcare
7
  - clinicalbert
8
  - symptom-checker
 
9
  license: apache-2.0
10
  datasets:
11
  - qilex/Symptom2Disease
@@ -39,8 +42,6 @@ It has been trained on a massive merged dataset of over **96,000 patient cases**
39
  | 2 | 0.3092 | 0.2852 |
40
  | 3 | 0.2526 | **0.2577** |
41
 
42
-
43
-
44
  The model achieves a final validation loss of **0.2577**, demonstrating high confidence and generalization capabilities across 115 disease classes.
45
 
46
  ## 🚀 How to Use (Python)
@@ -51,7 +52,7 @@ You can use this model directly with the Hugging Face `pipeline`.
51
  from transformers import pipeline
52
 
53
  # Load the pipeline
54
- classifier = pipeline("text-classification", model="YOUR_USERNAME/YOUR_MODEL_NAME", top_k=3)
55
 
56
  # Test with symptoms
57
  symptoms = "I have a severe headache, sensitivity to light, and I feel nauseous."
 
1
  ---
2
  language: en
3
+ pipeline_tag: text-classification
4
+ library_name: transformers
5
  tags:
6
  - medical
7
  - classification
8
  - healthcare
9
  - clinicalbert
10
  - symptom-checker
11
+ - text-classification
12
  license: apache-2.0
13
  datasets:
14
  - qilex/Symptom2Disease
 
42
  | 2 | 0.3092 | 0.2852 |
43
  | 3 | 0.2526 | **0.2577** |
44
 
 
 
45
  The model achieves a final validation loss of **0.2577**, demonstrating high confidence and generalization capabilities across 115 disease classes.
46
 
47
  ## 🚀 How to Use (Python)
 
52
  from transformers import pipeline
53
 
54
  # Load the pipeline
55
+ classifier = pipeline("text-classification", model="Iloriayomide/Symptom_Prediction", top_k=3)
56
 
57
  # Test with symptoms
58
  symptoms = "I have a severe headache, sensitivity to light, and I feel nauseous."