Xavier Fontaine
commited on
Commit
·
6db3e50
1
Parent(s):
553ea8c
Update README.md
Browse files
README.md
CHANGED
|
@@ -23,7 +23,7 @@ model-index:
|
|
| 23 |
metrics:
|
| 24 |
- type: f1
|
| 25 |
value: 0.804
|
| 26 |
-
name: micro-F1 score
|
| 27 |
- type: precision
|
| 28 |
value: 0.817
|
| 29 |
name: precision
|
|
@@ -33,4 +33,43 @@ model-index:
|
|
| 33 |
- type: accuracy
|
| 34 |
value: 0.859
|
| 35 |
name: accuracy
|
| 36 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
metrics:
|
| 24 |
- type: f1
|
| 25 |
value: 0.804
|
| 26 |
+
name: micro-F1 score
|
| 27 |
- type: precision
|
| 28 |
value: 0.817
|
| 29 |
name: precision
|
|
|
|
| 33 |
- type: accuracy
|
| 34 |
value: 0.859
|
| 35 |
name: accuracy
|
| 36 |
+
---
|
| 37 |
+
|
| 38 |
+
# ClinicalNER
|
| 39 |
+
|
| 40 |
+
## Model Description
|
| 41 |
+
|
| 42 |
+
This is a multilingual clinical NER model extracting DRUG, STRENGTH, FREQUENCY, DURATION, DOSAGE and FORM entities from a medical text.
|
| 43 |
+
|
| 44 |
+
## Evaluation Metrics on [MedNERF dataset](https://huggingface.co/datasets/Posos/MedNERF)
|
| 45 |
+
|
| 46 |
+
- Loss: 0.692
|
| 47 |
+
- Accuracy: 0.859
|
| 48 |
+
- Precision: 0.817
|
| 49 |
+
- Recall: 0.791
|
| 50 |
+
- micro-F1: 0.804
|
| 51 |
+
- macro-F1: 0.819
|
| 52 |
+
|
| 53 |
+
## Usage
|
| 54 |
+
|
| 55 |
+
```
|
| 56 |
+
from transformers import AutoModelForTokenClassification, AutoTokenizer
|
| 57 |
+
|
| 58 |
+
model = AutoModelForTokenClassification.from_pretrained("Posos/ClinicalNER")
|
| 59 |
+
tokenizer = AutoTokenizer.from_pretrained("Posos/ClinicalNER")
|
| 60 |
+
|
| 61 |
+
inputs = tokenizer("Take 2 pills every morning", return_tensors="pt")
|
| 62 |
+
outputs = model(**inputs)
|
| 63 |
+
```
|
| 64 |
+
|
| 65 |
+
## Citation information
|
| 66 |
+
|
| 67 |
+
```
|
| 68 |
+
@inproceedings{mednerf,
|
| 69 |
+
title = "Multilingual Clinical NER: Translation or Cross-lingual Transfer?",
|
| 70 |
+
author = "Gaschi, Félix and Fontaine, Xavier and Rastin, Parisa and Toussaint, Yannick",
|
| 71 |
+
booktitle = "Proceedings of the 5th Clinical Natural Language Processing Workshop",
|
| 72 |
+
publisher = "Association for Computational Linguistics",
|
| 73 |
+
year = "2023"
|
| 74 |
+
}
|
| 75 |
+
```
|