bert-event-extraction

Pytorch Solution of Event Extraction Task using BERT on ACE 2005 corpus

Prerequisites

1. Prepare ACE 2005 dataset.

2. Use nlpcl-lab/ace2005-preprocessing to preprocess ACE 2005 dataset in the same format as the data/sample.json. Then place it in the data directory as follows:

   ├── data
   │     └── test.json
   │     └── dev.json
   │     └── train.json
   │...
   

3. Install the packages.

   pip install pytorch==1.0 pytorch_pretrained_bert==0.6.1 numpy
   

Usage

Train

python train.py

Evaluation

python eval.py --model_path=latest_model.pt

Result

Performance

The performance of this model is low in argument classification even though pretrained BERT model was used. The model is currently being updated to improve the performance.

Reference

• Jointly Multiple Events Extraction via Attention-based Graph Information Aggregation (EMNLP 2018), Liu et al. [paper]
• lx865712528's EMNLP2018-JMEE repository [github]
• Kyubyong's bert_ner repository [github]