Skip to content

thunlp/Adv-ED

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

Adv-ED

Source code and dataset for NAACL 2019 paper "Adversarial Training for Weakly Supervised Event Detection".

Requirements

  • python == 3.6.3
  • pytorch == 0.4.1
  • numpy == 1.15.2
  • sklearn == 0.20.0
  • pytorch-pretrained-bert == 0.2.0

Data

Due to the licence issues, we cannot share the source ACE2005 dataset or the preprocessed data.

So we specify the data format in DataFormat.md and you can preprocess the data follow the format.

Run

Put the preprocessed .npy data files in the same directory as the codes.

For the BERT models, download the Bert_base_uncase model in ../../BERT_CACHE.

Run python train.py in corresponding directory to train the model.

If you want to tune the hyper parameters, see the constant.py and change the parameters defined in the file.

Cite

If the codes help you, please cite the following paper:

Adversarial Training for Weakly Supervised Event Detection. Xiaozhi Wang, Xu Han, Zhiyuan Liu, Maosong Sun, Peng Li. NAACL-HLT 2019.

About

Source code and dataset for NAACL 2019 paper "Adversarial Training for Weakly Supervised Event Detection".

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages