Commit
·
b5ea1fe
1
Parent(s):
e2a4b08
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,49 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
language: en
|
| 4 |
+
tags:
|
| 5 |
+
|
| 6 |
+
- adverse-drug-events
|
| 7 |
+
- twitter
|
| 8 |
+
- social-media-mining-for-health
|
| 9 |
+
- SMM4H
|
| 10 |
+
|
| 11 |
+
widget:
|
| 12 |
+
- text: "joints killing me now i have gone back up on the lamotrigine. sick of side effects. sick of meds. want my own self back. knackered today"
|
| 13 |
+
example_title: "Adverse Drug Event Detection - ADE"
|
| 14 |
+
- text: "bout to have a kick ass summer then it's time to get serious fer school #vyvanse #geekmode"
|
| 15 |
+
example_title: "Adverse Drug Event Detection - noADE"
|
| 16 |
---
|
| 17 |
+
|
| 18 |
+
## t2t-assert-ade-balanced
|
| 19 |
+
|
| 20 |
+
t2t-assert-ade-balanced is a text-to-text (**t2t**) adverse drug event (**ade**) detection model trained with over- and undersampled (balanced) English tweets reporting adverse drug events. It is trained as part of BOUN-TABI system for the Social Media Mining for Health (SMM4H) 2022 shared task. The system description paper has been accepted for publication in *Proceedings of the Seventh Social Media Mining for Health (#SMM4H) Workshop and Shared Task* and will be available soon. The source code has been released on GitHub at [https://github.com/gokceuludogan/boun-tabi-smm4h22](https://github.com/gokceuludogan/boun-tabi-smm4h22).
|
| 21 |
+
|
| 22 |
+
The model utilizes the T5 model and its text-to-text formulation. The inputs are fed to the model directly as a sentence/tweet, and receives the output "adverse event problem" or "healthy okay".
|
| 23 |
+
|
| 24 |
+
## Requirements
|
| 25 |
+
```
|
| 26 |
+
sentencepiece
|
| 27 |
+
transformers
|
| 28 |
+
```
|
| 29 |
+
|
| 30 |
+
## Usage
|
| 31 |
+
|
| 32 |
+
```python
|
| 33 |
+
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
| 34 |
+
tokenizer = AutoTokenizer.from_pretrained("yirmibesogluz/t2t-assert-ade-balanced")
|
| 35 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("yirmibesogluz/t2t-assert-ade-balanced")
|
| 36 |
+
predictor = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
|
| 37 |
+
predictor("joints killing me now i have gone back up on the lamotrigine. sick of side effects. sick of meds. want my own self back. knackered today")
|
| 38 |
+
```
|
| 39 |
+
|
| 40 |
+
## Citation
|
| 41 |
+
|
| 42 |
+
```bibtex
|
| 43 |
+
@inproceedings{uludogan-gokce-yirmibesoglu-zeynep-2022-boun-tabi-smm4h22,
|
| 44 |
+
title = "{BOUN}-{TABI}@{SMM4H}'22: Text-to-{T}ext {A}dverse {D}rug {E}vent {E}xtraction with {D}ata {B}alancing and {P}rompting",
|
| 45 |
+
author = "Uludo{\u{g}}an, G{\"{o}}k{\c{c}}e and Yirmibe{\c{s}}o{\u{g}}lu, Zeynep",
|
| 46 |
+
booktitle = "Proceedings of the Seventh Social Media Mining for Health ({\#}SMM4H) Workshop and Shared Task",
|
| 47 |
+
year = "2022",
|
| 48 |
+
}
|
| 49 |
+
```
|