prakharsinghAI commited on
Commit
2b71edf
·
verified ·
1 Parent(s): f94195b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +72 -0
README.md ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - medical
5
+ - ner
6
+ - named-entity-recognition
7
+ - healthcare
8
+ - i2b2
9
+ license: apache-2.0
10
+ datasets:
11
+ - i2b2-2018
12
+ metrics:
13
+ - f1
14
+ - precision
15
+ - recall
16
+ ---
17
+
18
+ # i2b2 2018 Medical NER Model
19
+
20
+ This model is fine-tuned for medical Named Entity Recognition (NER) using the i2b2 2018 dataset.
21
+
22
+ ## Model Details
23
+
24
+ - **Task**: Named Entity Recognition (NER)
25
+ - **Domain**: Medical/Healthcare
26
+ - **Dataset**: i2b2 2018
27
+ - **Model Type**: Token Classification
28
+
29
+ ## Usage
30
+
31
+ ```python
32
+ from transformers import pipeline
33
+
34
+ # Load the model
35
+ ner_pipeline = pipeline(
36
+ "ner",
37
+ model="prakharsinghAI/i2b2-ner-model",
38
+ aggregation_strategy="simple"
39
+ )
40
+
41
+ # Example usage
42
+ text = "Patient was prescribed aspirin 100mg twice daily for headache."
43
+ results = ner_pipeline(text)
44
+ print(results)
45
+ ```
46
+
47
+ ## Training Details
48
+
49
+ - **Dataset**: i2b2 2018 Medical NER
50
+ - **Task**: Token Classification
51
+ - **Labels**: Medical entities (Drug, Dosage, Route, etc.)
52
+
53
+ ## Performance
54
+
55
+ This model was trained on the i2b2 2018 dataset for medical named entity recognition tasks.
56
+
57
+ ## Citation
58
+
59
+ If you use this model, please cite the i2b2 2018 dataset:
60
+
61
+ ```bibtex
62
+ @article{krallinger2015chemdner,
63
+ title={The CHEMDNER corpus of chemicals and drugs and its annotation principles},
64
+ author={Krallinger, Martin and Rabal, Obdulia and Leitner, Florian and Vazquez, Miguel and Salgado, David and Lu, Zhiyong and Leaman, Robert and Lu, Yanan and Ji, Donghong and Lowe, Daniel M and others},
65
+ journal={Journal of cheminformatics},
66
+ volume={7},
67
+ number={1},
68
+ pages={1--17},
69
+ year={2015},
70
+ publisher={BioMed Central}
71
+ }
72
+ ```