资源算法bert-event-extraction

bert-event-extraction

2020-03-10 | |  39 |   0 |   0

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

MethodTrigger Classification (%)Argument Classification (%)
PrecisionRecallF1PrecisionRecallF1
JRNN66.073.069.354.256.755.5
JMEE76.371.373.766.854.960.3
This model (BERT base)63.471.167.748.534.140.0

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]


上一篇:BERT-Dureader

下一篇:bert-toxicity-classification

用户评价
全部评价

热门资源

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...