Downloads · 30 days
106
1% of all-time downloads
BENT-PubMedBERT-NER-Disease
pruas/BENT-PubMedBERT-NER-Disease
BENT-PubMedBERT-NER-Disease is a token classification model from pruas. Use it when you need labels on individual words, such as names. It is set up for transformers. The card lists the license as apache-2.0.
Named Entity Recognition (NER) model to recognize disease entities.
All-time downloads
20.8K
Public
Repo size
1.7 GB
Likes
7
Public
Repo makeup
Click a slice to open those files.
.pt871 MB · 67%
At a glance
- Task
- Token Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- bert
- Access
- Public
- Created
- Dec 13, 2022
- Updated
- Mar 2, 2024
- SHA
- e70d6a5b
Try a prompt
Model card
From the Hugging Face model README
Named Entity Recognition (NER) model to recognize disease entities.
Please cite our work:
@article{NILNKER2022,
title = {NILINKER: Attention-based approach to NIL Entity Linking},
journal = {Journal of Biomedical Informatics},
volume = {132},
pages = {104137},
year = {2022},
issn = {1532-0464},
doi = {https://doi.org/10.1016/j.jbi.2022.104137},
url = {https://www.sciencedirect.com/science/article/pii/S1532046422001526},
author = {Pedro Ruas and Francisco M. Couto},
}
PubMedBERT fine-tuned on the following datasets:
- NCBI Disease Corpus (train and dev sets)
- PHAEDRA (train, dev, test sets): entity type "Disorder"
- Corpus for Disease Names and Adverse Effects (train, dev, test sets): entity types "DISEASE", "ADVERSE"
- RareDis corpus (train, dev, test sets): entity types "DISEASE", "RAREDISEASE", "SYMPTOM"
- CoMAGC (train, dev, test sets): entity type "cancer_term"
- PGxCorpus (train, dev, test sets):
- miRNA-Test-Corpus (train, dev, test sets): entity type "Diseases"
- BC5CDR (train and dev sets): entity type "Disease"
- Mantra (train, dev, test sets): entity type "DISO"
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- License
- apache-2.0
- Languages
- en
- Created
- Dec 13, 2022
- Updated
- Mar 2, 2024