Update README.md
Browse files
README.md
CHANGED
|
@@ -1,11 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
|
| 3 |
-
language:
|
| 4 |
-
- en
|
| 5 |
-
tags:
|
| 6 |
-
- Token Classification
|
| 7 |
-
co2_eq_emissions: 0.0279399890043426
|
| 8 |
-
widget:
|
| 9 |
- text: >-
|
| 10 |
CASE: A 28-year-old previously healthy man presented with a 6-week history
|
| 11 |
of palpitations. The symptoms occurred during rest, 2–3 times per week,
|
|
@@ -29,38 +23,3 @@ widget:
|
|
| 29 |
Pathological examination revealed that the tumour also extensively involved
|
| 30 |
the lower uterine segment.
|
| 31 |
example_title: example 3
|
| 32 |
-
datasets:
|
| 33 |
-
- tner/bc5cdr
|
| 34 |
-
- commanderstrife/jnlpba
|
| 35 |
-
- bc2gm_corpus
|
| 36 |
-
- drAbreu/bc4chemd_ner
|
| 37 |
-
- linnaeus
|
| 38 |
-
- chintagunta85/ncbi_disease
|
| 39 |
-
---
|
| 40 |
-
|
| 41 |
-
## About the Model
|
| 42 |
-
An English Named Entity Recognition model, trained on Maccrobat to recognize the bio-medical entities (107 entities) from a given text corpus (case reports etc.). This model was built on top of distilbert-base-uncased
|
| 43 |
-
|
| 44 |
-
- Dataset: Maccrobat https://figshare.com/articles/dataset/MACCROBAT2018/9764942
|
| 45 |
-
- Carbon emission: 0.0279399890043426 Kg
|
| 46 |
-
- Training time: 30.16527 minutes
|
| 47 |
-
- GPU used : 1 x GeForce RTX 3060 Laptop GPU
|
| 48 |
-
|
| 49 |
-
Checkout the tutorial video for explanation of this model and corresponding python library: https://youtu.be/xpiDPdBpS18
|
| 50 |
-
|
| 51 |
-
## Usage
|
| 52 |
-
The easiest way is to load the inference api from huggingface and second method is through the pipeline object offered by transformers library.
|
| 53 |
-
```python
|
| 54 |
-
from transformers import pipeline
|
| 55 |
-
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
| 56 |
-
|
| 57 |
-
tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all")
|
| 58 |
-
model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
|
| 59 |
-
|
| 60 |
-
pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple") # pass device=0 if using gpu
|
| 61 |
-
pipe("""The patient reported no recurrence of palpitations at follow-up 6 months after the ablation.""")
|
| 62 |
-
```
|
| 63 |
-
|
| 64 |
-
## Author
|
| 65 |
-
This model is part of the Research topic "AI in Biomedical field" conducted by Deepak John Reji, Shaina Raza. If you use this work (code, model or dataset), please star at:
|
| 66 |
-
> https://github.com/dreji18/Bio-Epidemiology-NER
|
|
|
|
| 1 |
---
|
| 2 |
+
l
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
- text: >-
|
| 4 |
CASE: A 28-year-old previously healthy man presented with a 6-week history
|
| 5 |
of palpitations. The symptoms occurred during rest, 2–3 times per week,
|
|
|
|
| 23 |
Pathological examination revealed that the tumour also extensively involved
|
| 24 |
the lower uterine segment.
|
| 25 |
example_title: example 3
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|