Update README.md
Browse files
README.md
CHANGED
@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
18 |
|
19 |
# clinical-ner
|
20 |
|
21 |
-
This model is a fine-tuned version of [microsoft/deberta-v3-base](https://huggingface.co/microsoft/deberta-v3-base) on the
|
22 |
It achieves the following results on the evaluation set:
|
23 |
- Loss: 0.8058
|
24 |
- Precision: 0.5786
|
@@ -55,6 +55,22 @@ The following hyperparameters were used during training:
|
|
55 |
- num_epochs: 45
|
56 |
- mixed_precision_training: Native AMP
|
57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
### Training results
|
59 |
|
60 |
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|
|
|
18 |
|
19 |
# clinical-ner
|
20 |
|
21 |
+
This model is a fine-tuned version of [microsoft/deberta-v3-base](https://huggingface.co/microsoft/deberta-v3-base) on the Medical dataset.
|
22 |
It achieves the following results on the evaluation set:
|
23 |
- Loss: 0.8058
|
24 |
- Precision: 0.5786
|
|
|
55 |
- num_epochs: 45
|
56 |
- mixed_precision_training: Native AMP
|
57 |
|
58 |
+
### Python Code:
|
59 |
+
```python
|
60 |
+
# Use a pipeline as a high-level helper
|
61 |
+
from transformers import pipeline
|
62 |
+
pipe = pipeline("token-classification", model="Clinical-AI-Apollo/Medical-NER", aggregation_strategy='simple')
|
63 |
+
result = pipe('45 year old woman diagnosed with CAD')
|
64 |
+
|
65 |
+
|
66 |
+
|
67 |
+
# Load model directly
|
68 |
+
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
69 |
+
|
70 |
+
tokenizer = AutoTokenizer.from_pretrained("Clinical-AI-Apollo/Medical-NER")
|
71 |
+
model = AutoModelForTokenClassification.from_pretrained("Clinical-AI-Apollo/Medical-NER")
|
72 |
+
```
|
73 |
+
|
74 |
### Training results
|
75 |
|
76 |
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|