Update README.md
Browse files
README.md
CHANGED
|
@@ -6,7 +6,19 @@ tags:
|
|
| 6 |
base_model: distilbert/distilbert-base-cased
|
| 7 |
model-index:
|
| 8 |
- name: distilbert-finetuned-medical-diagnosis
|
| 9 |
-
results:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
datasets:
|
| 11 |
- ninaa510/diagnosis-text
|
| 12 |
language:
|
|
@@ -56,4 +68,4 @@ The following hyperparameters were used during training:
|
|
| 56 |
- Transformers 4.41.0
|
| 57 |
- TensorFlow 2.15.0
|
| 58 |
- Datasets 2.19.1
|
| 59 |
-
- Tokenizers 0.19.1
|
|
|
|
| 6 |
base_model: distilbert/distilbert-base-cased
|
| 7 |
model-index:
|
| 8 |
- name: distilbert-finetuned-medical-diagnosis
|
| 9 |
+
results:
|
| 10 |
+
- task:
|
| 11 |
+
type: text-classification
|
| 12 |
+
name: Text classification
|
| 13 |
+
dataset:
|
| 14 |
+
type: ninaa510/diagnosis-text
|
| 15 |
+
name: Symptoms and diseases for classification
|
| 16 |
+
split: test
|
| 17 |
+
metrics:
|
| 18 |
+
- type: accuracy # Required. Example: wer. Use metric id from https://hf.co/metrics
|
| 19 |
+
value: 58.68 # Required. Example: 20.90
|
| 20 |
+
name: Accuracy # Optional. Example: Test WER
|
| 21 |
+
|
| 22 |
datasets:
|
| 23 |
- ninaa510/diagnosis-text
|
| 24 |
language:
|
|
|
|
| 68 |
- Transformers 4.41.0
|
| 69 |
- TensorFlow 2.15.0
|
| 70 |
- Datasets 2.19.1
|
| 71 |
+
- Tokenizers 0.19.1
|