Xavier Fontaine commited on
Commit
6db3e50
1 Parent(s): 553ea8c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -2
README.md CHANGED
@@ -23,7 +23,7 @@ model-index:
23
  metrics:
24
  - type: f1
25
  value: 0.804
26
- name: micro-F1 score # Optional. Example: Test WER
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
+ ```