OTE-MTL - Multi-Task Learning for Opinion Triplet Extraction
- Code and preprocessed dataset for EMNLP 2020 Findings paper titled "A Multi-task Learning Framework for Opinion Triplet Extraction"
- Chen Zhang, Qiuchi Li, Dawei Song and Benyou Wang.
We folk this repository from the OTE-MTL repository and evaluate the OTE-MTL model on the ASOTE task.
sh run.sh 4 train.py --model ote --dataset rest14 > rest14.log 2>&1 &
sh run.sh 5 train.py --model ote --dataset laptop14 > laptop14.log 2>&1 &
sh run.sh 6 train.py --model ote --dataset rest15 > rest15.log 2>&1 &
sh run.sh 7 train.py --model ote --dataset rest16 > rest16.log 2>&1 &
- Feb. 20th, 2021. As is pointed out in our paper, we have noted that datav1 used in https://arxiv.org/abs/1911.01616 is rather incomplete and have corrected their mistakes. That is, the data used for our experiments is similar to datav2. However, as is requested by some users and in case of any inconsistencies between our data and datav2, we decide to support the test of our model on datav2. You could just run our model on datav2 with just an additional argument --v2.
- Python 3.6
- PyTorch 1.0.0
- numpy 1.15.4
- Download pretrained GloVe embeddings with this link and extract
glove.840B.300d.txt
intoglove/
. - Train with command, optional arguments could be found in train.py, --v2 denotes whether test on datav2
python train.py --model ote --dataset rest14 [--v2]
- Infer with infer.py
An overview of the task opinion triplet extraction (OTE) is given below
OTE is solving the same task proposed in https://arxiv.org/abs/1911.01616. While our work focuses on extracting (aspect term, opinion term, sentiment) opinion triplets (OTs), they extract (aspect term-sentiment pair, opinion term)s. Owing to the minor difference lying in formulations, two drawbacks in the latter formulation are presented: (i) sentiments are determined without accessing opinion terms, (ii) conflicting opinions expressed towards an aspect cannot be predicted.
If you use the code in your paper, please kindly star this repo and cite our paper
@inproceedings{zhang-etal-2020-multi,
title = "A Multi-task Learning Framework for Opinion Triplet Extraction",
author = "Zhang, Chen and
Li, Qiuchi and
Song, Dawei and
Wang, Benyou",
booktitle = "Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing: Findings",
month = nov,
year = "2020",
address = "Online",
publisher = "Association for Computational Linguistics",
url = "https://www.aclweb.org/anthology/2020.findings-emnlp.72",
pages = "819--828",
}
- For any issues or suggestions about this work, don't hesitate to create an issue or directly contact me via gene_zhangchen@163.com !