IARE

Integrated attention using entity type constraints and relation augmentation for distantly supervised relation extraction

DATA

• We use dataset from https://drive.google.com/file/d/12ROz5GkSEl5Ka9a1uuftPqcdpM6aFhGy/view?usp=sharing, and extract them in data/

Requirements

• Python 3.6+
• Pytorch 1.4.0+
• PyTorch Geometric (see https://github.com/rusty1s/pytorch_geometric for detail)
• Transformers

Training & Evaluation

• python train.py --config 520K_PCNN_GCN.yaml
• python eval.py --config 520K_PCNN_GCN.yaml

Acknowledgement

This repo is mainly based on CGRE https://github.com/tmliang/CGRE