Skip to content

Code for "Global and Local Hierarchy-aware Contrastive Framework for Hierarchical Implicit Discourse Relation Recognition (ACL 2023)"

License

Notifications You must be signed in to change notification settings

YJiangcm/GOLF_for_IDRR

Repository files navigation

GOLF_for_IDRR

Global and Local Hierarchy-aware Contrastive Framework for Hierarchical Implicit Discourse Relation Recognition

Requirements

First, install PyTorch by following the instructions from the official website. To faithfully reproduce our results, please use the correct 1.10.1 version corresponding to your platforms/CUDA versions. PyTorch version higher than 1.10.1 should also work.

Then run the following script to install the remaining dependencies,

pip install -r requirements.txt

Data Preparation before Training

  1. Download the PDTB2.0 dataset, put it under /raw/
  2. Run the following script for data preprocessing,
python3 preprocess.py

Train, Evaluate, and Test

Run the following script for training, evaludating, and testing,

python3 run.py

Ciation

Please cite our paper by:

@misc{jiang2022promcse,
      title={Improved Universal Sentence Embeddings with Prompt-based Contrastive Learning and Energy-based Learning}, 
      author={Yuxin Jiang, Linhan Zhang and Wei Wang},
      year={2022},
      eprint={2203.06875},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

About

Code for "Global and Local Hierarchy-aware Contrastive Framework for Hierarchical Implicit Discourse Relation Recognition (ACL 2023)"

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages