fyubang / copy_re

Release for acl18 paper "Extracting Relational Facts by an End-to-End Neural Model with Copy Mechanism"

This code is for ACL2018 paper "Extracting Relational Facts by an End-to-End Neural Model with Copy Mechanism"

Environment

• python2.7
• requirements.txt

Data

You need to modify the data path in const.py before running the code. The pre-processed data is released.

WebNLG:

• dataset
• pre-trained word embedding

NYT:

• dataset
• pre-trained word embedding

Train

python main.py -c config.json -t 0 -cell lstm

The cell can be "gru" or "lstm". You can specify the GPU card number by "-g". For exampe, "python main.py -c config.json -t 0 -cell lstm -g 0".

Test or Valid

You need to set the epochs of the model in main.py first. Then run the following commands: python main.py -c config.json -t 1 -cell lstm

"-t 1" means test and "-t 2" means valid.