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 |
+
```
|