Downloads · 30 days
19
2% of all-time downloads
treatment-disease-NER
jnferfer/treatment-disease-NER
treatment-disease-NER is a token classification model from jnferfer. Use it when you need labels on individual words, such as names. It is set up for transformers.
This is a model to detect treatment and disease mentions in texts from health domains. The dataset used for training consists of PubMed's abstracts and tweets with disease mentions, which are publicly available. It ha…
All-time downloads
1.1K
Public
Repo size
1.7 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin436 MB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jan 30, 2023
- Updated
- Feb 6, 2023
- SHA
- 2c0b4d14
Try a prompt
Model card
From the Hugging Face model README
This is a model to detect treatment and disease mentions in texts from health domains. The dataset used for training consists of PubMed's abstracts and tweets with disease mentions, which are publicly available. It has been semi-automatically labeled with a set of regex rules made ad hoc to find treatment-disease links.
The label t is used for treatment (medication, procedures, etc.) and d for disease mentions.
The current F1 score (Seqeval) is 0.91
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Jan 30, 2023
- Updated
- Feb 6, 2023