This is the prototype code for Relation Extraction Downstream Network of pre-trained language model, supporting our paper Downstream Model Design of Pre-trained Language Model for Relation Extraction Task.
Part of this code are revised based on OPENNRE.
You can get Datasets from OPENNRE and WebNLG 2.0
Set your own paths in example/configs.py, including pre-trained model path, root path of data and output name. Run example/redn_trainer.py with args dataset and mode. dataset can be nyt10, semeval or webnlg. mode can be t for training and e for evaluation. For example ,to train SemEval, try
python redn_trainer semeval t
If you are not able to run these codes, you can also check all the logs in ./logs.