Skip to content

Tag-enhanced Tree-Structured NN for discourse relation classification.

License

Notifications You must be signed in to change notification settings

yizhongw/TagNN-PDTB

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TagNN-PDTB

This repo contains the code for a state-of-the-art discourse relation classifier, trained and tested on PDTB V2.

Here We implemented the Tree-LSTM and Tree-GRU models and enhanced them with pos-tag information. You can find more details in our paper at IJCNLP 2017: Tag-Enhanced Tree-Structured Neural Networks for Implicit Discourse Relation Classification.

Data

We trained and and tested our models on PDTB V2 dataset. You should download them and convert the PDTB into a "pipe" delimited file using the built-in tool in the dataset.

Usage

To run the program, you can use the following command for preprocessing the data, prepare the formatted dataset and train the models:

python main.py [--preprocess] [--prepare] [--train]

For detailed usage, run python main.py -h to see the list of options.

License

MIT license

About

Tag-enhanced Tree-Structured NN for discourse relation classification.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages