Skip to content

AAAI20 "CopyMTL: Copy Mechanism for Joint Extraction of Entities and Relations with Multi-Task Learning"

Notifications You must be signed in to change notification settings

WindChimeRan/CopyMTL

Repository files navigation

UPDATE: This paper is part of the our EMNLP2020 findings: paper code.

  • Model bias: CopyMTL suffers from the exposure bias problem, which can be solved by our Seq2UMTree.
  • Data bias: NYT dataset is overfitted by SoTA models. This is because 90% test triplets reoccured in the training data.
  • We release OpenJERE toolkit, including multiple baselines and datasets. CopyMTL can be found here!

CopyMTL: Copy Mechanism for Joint Extraction of Entities and Relations with Multi-Task Learning

Paper accepted by AAAI-2020

This is a followup paper of "Extracting Relational Facts by an End-to-End Neural Model with Copy Mechanism" ACL2018 CopyRE

Environment

python3

pytorch 0.4.0 -- 1.3.1

Modify the Data path

This repo initially contain webnlg, you can run the code directly. NYT dataset need to be downloaded and to be placed in proper path. see const.py.

The pre-processed data is avaliable in:

WebNLG dataset: https://drive.google.com/open?id=1zISxYa-8ROe2Zv8iRc82jY9QsQrfY1Vj

NYT dataset: https://drive.google.com/open?id=10f24s9gM7NdyO3z5OqQxJgYud4NnCJg3

Run

python main.py --gpu 0 --mode train --cell lstm --decoder_type one

python main.py --gpu 0 --mode test --cell lstm --decoder_type one

About

AAAI20 "CopyMTL: Copy Mechanism for Joint Extraction of Entities and Relations with Multi-Task Learning"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published